Unisci gli elenchi per valore

Question

Esiste un modo efficace per unire due elenchi di tuple in python, in base a un valore comune. Attualmente, sto facendo la seguente:

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) 

rendimenti:

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

Anche se questo codice funziona perfettamente bene per le piccole liste, è incredibilmente lenta per liste più lunghe con milioni di dischi. C'è un modo più efficiente per scrivere questo?

MODIFICA I numeri nella prima colonna sono univoci.

EDIT Modificato il codice di @John Kugelman leggermente. Aggiunto un get(), nel caso in cui il dizionario nomi non ha una chiave corrispondente nel dizionario occupazione:

>>>> 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

Uso dizionari invece di liste piatte.

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

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

Se avete bisogno di convertirli in questo formato, si può fare:

>>> {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'} 

Sarebbe quindi essere un pezzo di torta per unire i due.

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']] 

Se si desidera che i dati in ordine di come appare nei nomi allora avete bisogno di utilizzare un OrderedDict come dicts normali non sono ordinati.

è anche possibile aggiungere i dati del ciclo creando le tuple desiderati poi basta chiamare od.values ​​per ottenere l'elenco di tuple:

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')]