2010-10-09 24 views
59

Ho una lista Python di dizionari, come segue:Verificare se il valore esiste già nella lista dei dizionari?

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

mi piacerebbe verificare se un dizionario con una particolare chiave/valore esiste già nella lista, come segue:

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

risposta

131

Ecco un modo per farlo:

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

La parte tra parentesi è un generatore di espressione che restituisce True per ogni dictiona ry che ha la coppia chiave-valore che stai cercando, altrimenti False.


Se la chiave potrebbe anche mancare il codice di cui sopra può dare un KeyError. È possibile risolvere questo problema utilizzando get e fornendo un valore predefinito.

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

Very nice. Grazie! – AP257

+0

Incredibile sintassi one-liner, sono stato così duro per questo! Sono curioso che dove nei documenti Python che ci mostrano possiamo effettivamente mettere l'operazione di 'for' prima del 'for'? – sylye

+1

L'ho trovato, si chiama List Comprehensions https://docs.python.org/2/whatsnew/2.0.html?highlight=comprehensions – sylye

2

Forse una funzione in questo senso è quello che stai dopo:

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

Forse questo aiuta:

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) 
Problemi correlati