Объединить списки по значению

Question

Есть ли эффективный способ объединения двух списков кортежей в python, основанных на общем значении. В настоящее время я делаю следующее:

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) 

возвращается:

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

Хотя этот код работает прекрасно для небольших списков, это невероятно медленно для длинных списков с миллионами записей. Есть ли более эффективный способ написать это?

EDIT Номера в первом столбце уникальны.

EDIT Измененный код @John Kugelman. Добавленный Get(), только в том случае, словарь имен не имеет соответствующий ключ в словаре оккупационного:

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

Используйте словари вместо плоских списков.

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

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

Если вам нужно преобразовать их в этот формат, вы можете сделать:

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

Было бы затем быть кусок пирога, чтобы объединить два.

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

Если вы хотите, чтобы данные, заказанные, как представляется, в именах, то вам нужно использовать OrderedDict, как нормальные dicts неупорядоченны.

Вы также можете добавить данные в цикле создания требуемого кортежи, то просто звоню od.values, чтобы получить список кортежей:

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