Encontrar um elemento de uma lista em um dicionário e voltar a chave

votos
1

Se eu tivesse someDictionarydefinido como

{'1': 33, '2': 44, '3': 55, '4': 10, '5': 66, '6': 60}

e someListdefinido como

[51, 52, 53, 54, 55, 56]

como posso encontrar a chave em someDictionaryque coincide com um elemento em someList? Por agora, eu estou supondo que não haverá mais do que um jogo.

Eu acho que seria algo como isto:

for k, v in someDictionary.items():
    if v == someList:
        myAnswer = k

Dado o código acima, myAnswerseria3

A segunda parte que eu preciso fazer é, se someList não contém um elemento que está em someDictionary, encontrar o valor em someDictionarymaior do que (mas mais próximo a) o último elemento someListsomeList[-1]

Nesse caso, myAnswer seria 6

Publicado 20/10/2018 em 14:12
fonte usuário
Em outras línguas...                            


8 respostas

votos
2

Primeira parte, Encontre a lista de chaves que têm um valor dicionário na lista.

someDictionary = {'1': 33, '2': 44, '3': 55, '4': 10, '5': 66, '6': 60}
someList = [51, 52, 53, 54, 55, 56]

res = [key for key, value in someDictionary.items() if value in someList]

segunda parte, se nenhum resultado da primeira parte, encontrar a chave com o mais próximo maior valor (continuação):

if not res:
    res = min ([(value, key) for key, value in someDictionary.items() if value > max(someList) ])[1]
Respondeu 20/10/2018 em 14:20
fonte usuário

votos
1
myAnswer = ''
closest_to = someList[-1]
closest_diff = 0
for k in someDictionary:
    if someDictionary[k] in someList:
         myAnswer = k
         break; # stop the loop when exact match found
    diff = someDictionary[k] - closest_to
    if diff > 0 and diff < closest_diff:
         closest_diff = diff
         myAnswer = k # this would save key of larger but closest number to last in someList
Respondeu 20/10/2018 em 14:24
fonte usuário

votos
1

Primeira lista utilização compreensão para reunir todas as chaves cujos valores estão em lst, se não houver correspondência, em seguida, filtrar para chaves cujo valor é maior do que lst[-1]. Depois de tipo-los com base no valor abs da diferença dos valores de chaves e o último item lste levar o 0índice, o item mais próximo.

dct = {'1': 33, '2': 44, '3': 55, '4': 10, '5': 66, '6': 60}
lst = [51, 52, 53, 54, 55, 56]

my_answer = [int(i) for i in dct if dct[i] in lst]
if my_answer:
    print(*my_answer) # => 3
else:
    close = [i for i in dct if int(dct[i]) > max(lst)]
    closest = sorted(close, key=lambda x: abs(dct.get(x) - lst[-1]))[0]
    print(closest) # => 6
Respondeu 20/10/2018 em 14:25
fonte usuário

votos
1

Como posso encontrar a chave em someDictionaryque coincide com um elemento em someList?

Um dicionário é usado chaves de mapa de valores. O reverso não é possível em O (1) hora. Mas você pode interagir em tempo O ( n ) tempo até que você pode combinar um elemento da lista com um valor de dicionário.

Você pode usar um simples forcircuito para este, seja repetindo seu dicionário ou lista.

d = {'1': 33, '2': 44, '3': 55, '4': 10, '5': 66, '6': 60}
L = [51, 52, 53, 54, 55, 56]

def find_key_dict(d, L):
    L_set = set(L)  # convert to set for O(1) lookup
    for k, v in d.items():
        if v in L_set:
            return k

def find_key_list(d, L):
    d_rev = {v: k for k, v in d.items()}  # reverse dict for value -> key map
    for i in L:
        if i in d_rev:
            return d_rev[i]

find_key_dict(d, L)  # '3'
find_key_list(d, L)  # '3'

Também é possível reescrever estas funções como gerador de expressões de 1 linha com next, mas isso não será necessariamente mais eficiente.

A segunda parte que eu preciso fazer é, se someListnão contém um elemento que está em someDictionary, encontrar o valor em someDictionary maior do que (mas mais próximo a) o último elementosomeList

Você pode escrever uma função semelhante utilizando uma for... else...construção com min:

def find_key_dict(d, L):
    L_set = set(L)  # convert to set for O(1) lookup
    for k, v in d.items():
        if v in L_set:
            return k
    else:
        def min_func(x):
            diff = x[1] - L[-1]
            return diff <= 0, diff
        return min(d.items(), key=min_func)[0]

find_key_dict(d, L)  # '6'
Respondeu 20/10/2018 em 14:27
fonte usuário

votos
1

Para a primeira pergunta, você está apenas usando o operador errado para verificar se o valor está em someList

for k, v in someDictionary.items():
    if v in someList:
        myAnswer = k

Sobre a segunda pergunta que você pode estender o código anterior, desta forma

for k, v in someDictionary.items():
     if v in someList:
         myAnswer = k
         break
else:
     myAnswer = None
     for k, v in someDictionary.items():
         if v > someList[-1] and (myAnswer is None or v < someDictionary[myAnswer]):
             myAnswer = k
Respondeu 20/10/2018 em 14:33
fonte usuário

votos
3

Parece que você quer um bidict

>>> from bidict import bidict  # pip install bidict
>>> d = bidict({'1': 33, '2': 44, '3': 55, '4': 10, '5': 66, '6': 60})
>>> d['3'] 55
>>> d.inv[55] '3'

Isto permite que o (1) pesquisas em um ou outro sentido. Agora, você pode fazer um loop sobre someListe verificar se um elemento é in d.inveficiente.

Respondeu 20/10/2018 em 14:38
fonte usuário

votos
1

Se o seu dicionário e lista são grandes, e especialmente se você tiver muitas listas para testar contra o mesmo dicionário, pode valer a pena a preparar os seus dados, a fim de ter uma execução mais rápida.

O pior operação em __init__é a classificação, que é O (n * log (n)). Ele nos permite usar bisectpara encontrar o valor mais próximo em seu último caso em O (log (n)).

from bisect import bisect


class FindInDict:
    def __init__(self, someDictionary):
        self.dict_by_values = {val: key for key, val in someDictionary.items()}
        self.dict_values_set = set(sorted_values)
        self.sorted_values = sorted(someDictionary.values())

    def find(self, someList):
        common = set(someList).intersection(self.dict_values_set)
        if common:
            key = self.dict_by_values[common.pop()]
        else:
            closest_value = self.sorted_values[bisect(self.sorted_values, someList[-1])]
            key = self.dict_by_values[closest_value]
        return key



someDictionary = {'1': 33, '2': 44, '3': 55, '4': 10, '5': 66, '6': 60}    
finder = FindInDict(someDictionary)

finder.find([51, 52, 53, 54, 55, 56])
# 3

finder.find([51, 52, 53, 54, 56])  # no common value
# 6
Respondeu 20/10/2018 em 15:08
fonte usuário

votos
1

Aqui está algo que lida com ambas as partes da sua pergunta e deve ser fácil de modificar para lidar com quaisquer traquinas "Edge-casos" que podem aparecer:

someDictionary = {'1': 33, '2': 44, '3': 55, '4': 10, '5': 66, '6': 60}
someList = [51, 52, 53, 54, 55, 56]

myAnswer = [key for key in someDictionary if someDictionary[key] in someList]
if not myAnswer:
    diffs = {dict_key: dict_value-someList[-1]
                for dict_key, dict_value in someDictionary.items()
                    if dict_value-someList[-1] > 0}
    myAnswer = [min(diffs, key=diffs.get)]  # Key of value with minimum
                                            # (positive) difference
print(myAnswer)
Respondeu 20/10/2018 em 16:00
fonte usuário

Cookies help us deliver our services. By using our services, you agree to our use of cookies. Learn more