2010-10-09 17 views
59

Mam listę Pythona słowników, co następuje:Sprawdź, czy wartość już istnieje na liście słowników?

a = [ 
    {'main_color': 'red', 'second_color':'blue'}, 
    {'main_color': 'yellow', 'second_color':'green'}, 
    {'main_color': 'yellow', 'second_color':'blue'}, 
] 

Chciałbym sprawdzić czy słownikiem ze szczególnym klucz/wartość już istnieje na liście, co następuje:

// is a dict with 'main_color'='red' in the list already? 
// if not: add item 

Odpowiedz

131

Oto jeden ze sposobów, aby to zrobić:

if not any(d['main_color'] == 'red' for d in a): 
    # does not exist 

część w nawiasach jest wyrażeniem, które zwraca True generator dla każdej dictiona ry, który ma parę klucz-wartość, której szukasz, inaczej: False.


Jeśli klucz może być również brak powyższy kod może dać KeyError. Możesz to naprawić, używając get i podając wartość domyślną.

if not any(d.get('main_color', None) == 'red' for d in a): 
    # does not exist 
+1

bardzo miłe. Dzięki! – AP257

+0

Niesamowita składnia jednego liniowca, wyglądam tak ciężko! Ciekaw jestem, skąd w dokumentach Pythona, które pokazują nam, że rzeczywiście możemy umieścić operację "za" przed "za"? – sylye

+1

Znalazłem to, nazywa się zrozumienie list https://docs.python.org/2/whatsnew/2.0.html?highlight=comprehensions – sylye

2

Być może funkcja wzdłuż tych linii jest to, co jesteś po:

def add_unique_to_dict_list(dict_list, key, value): 
    for d in dict_list: 
    if key in d: 
     return d[key] 

    dict_list.append({ key: value }) 
    return value 
4

Może to pomoże:

a = [{ 'main_color': 'red', 'second_color':'blue'}, 
    { 'main_color': 'yellow', 'second_color':'green'}, 
    { 'main_color': 'yellow', 'second_color':'blue'}] 

def in_dictlist((key, value), my_dictlist): 
    for this in my_dictlist: 
     if this[key] == value: 
      return this 
    return {} 

print in_dictlist(('main_color','red'), a) 
print in_dictlist(('main_color','pink'), a) 
Powiązane problemy