Przyłącz listy według wartości

Question

Czy istnieje skuteczny sposób scalania dwóch list krotek w python, na podstawie wspólnej wartości. Obecnie robię co następuje:

name = [ 
     (9, "John", "Smith"), 
     (11, "Bob", "Dobbs"), 
     (14, "Joe", "Bloggs") 
     ] 

occupation = [ 
       (9, "Builder"), 
       (11, "Baker"), 
       (14, "Candlestick Maker") 
       ] 

name_and_job = [] 

for n in name: 
    for o in occupation: 
     if n[0] == o[0]: 
      name_and_job.append((n[0], n[1], n[2], o[1])) 


print(name_and_job) 

powraca:

[(9, 'John', 'Smith', 'Builder'), (11, 'Bob', 'Dobbs', 'Baker'), (14, 'Joe', 'Bloggs', 'Candlestick Maker')] 

Podczas gdy ten kod działa perfekcyjnie dla małych listach, to niezwykle wolno przez dłuższy list z milionami rekordów. Czy istnieje skuteczniejszy sposób, aby to napisać?

EDYCJA Numery w pierwszej kolumnie są niepowtarzalne.

EDYTOWANIE Zmodyfikowano kod QR dla Jana Kugelmana. Dodano get(), na wypadek gdyby słownik nazw nie zawierał pasującego klucza w słowniku zawodów:

>>>> names_and_jobs = {id: names[id] + (jobs.get(id),) for id in names} 
>>>> print(names_and_jobs) 
{9: ('John', 'Smith', None), 11: ('Bob', 'Dobbs', 'Baker'), 14: ('Joe', 'Bloggs', 'Candlestick Maker')} 

Answer 1

Używaj słowników zamiast płaskich list.

names = { 
    9: ("John", "Smith"), 
    11: ("Bob", "Dobbs"), 
    14: ("Joe", "Bloggs") 
} 

jobs = { 
    9: "Builder", 
    11: "Baker", 
    14: "Candlestick Maker" 
} 

Jeśli trzeba konwertować je do tego formatu, można zrobić:

>>> {id: (first, last) for id, first, last in name} 
{9: ('John', 'Smith'), 11: ('Bob', 'Dobbs'), 14: ('Joe', 'Bloggs')} 
>>> {id: job for id, job in occupation} 
{9: 'Builder', 11: 'Baker', 14: 'Candlestick Maker'} 

Byłoby wtedy być bułka z masłem, aby połączyć dwa.

names_and_jobs = {id: names[id] + (jobs[id],) for id in names} 

Answer 2

from collections import OrderedDict 
from itertools import chain 

od = OrderedDict() 


for ele in chain(name,occupation): 
    od.setdefault(ele[0], []).extend(ele[1:]) 


print([[k]+val for k,val in od.items()]) 

[[9, 'John', 'Smith', 'Builder'], [11, 'Bob', 'Dobbs', 'Baker'], [14, 'Joe', 'Bloggs', 'Candlestick Maker']] 

Jeśli chcesz dane zamówionego przez jaki pojawia się w nazwach to trzeba użyć OrderedDict jako normalne dicts są nieuporządkowane.

Można również dodać dane w pętli tworzenia pożądanych krotki wtedy właśnie dzwoni od.values ​​aby uzyskać listę krotek:

from collections import OrderedDict 
from itertools import chain 

od = OrderedDict() 

for ele in chain(name, occupation): 
    k = ele[0] 
    if k in od: 
     od[k] = od[k] + ele[1:] 
    else: 
     od[k] = ele 

print(od.values()) 
[(9, 'John', 'Smith', 'Builder'), (11, 'Bob', 'Dobbs', 'Baker'), (14, 'Joe', 'Bloggs', 'Candlestick Maker')]