Создание словаря массивов с двумя списками с Python

Question

У меня есть особая проблема, которая меня озадачила. Предположим, у меня есть следующие два списка:

x = ["A","B","C","D","E"] 
y = [1,2,3,2,1] 

x и y имеют отношения. Связь привязана по индексу. То есть «A» относится к 1, «B», относящемуся к 2, «C», относящимся к 3 и так далее.

То, что я пытаюсь сделать, это создать отношение ключевого значения, когда уникальные элементы в y являются ключами, и каждый ключ имеет список, содержащий буквы, относящиеся к ключу, как упоминалось ранее. Я попытался сделать следующее:

mapping = dict(zip(y,x)) 
{1: 'E', 2: 'D', 3: 'C'} 

Это перезаписывает предыдущую букву. Я хотел бы иметь возможность вернуть следующее:

{1:['A','E'], 2:['B','D'], 3:['C']} 

У кого-нибудь есть умное решение? Предпочтительно без itertools.

Answer 1

Вы можете использовать setdefault

x = ["A","B","C","D","E"] 
y = [1,2,3,2,1] 
d = {} 
for i,j in zip(y,x): 
    d.setdefault(i, []).append(j) 
print d 

Выход:

{1: ['A', 'E'], 2: ['B', 'D'], 3: ['C']} 

Answer 2

defaultdict мои предпочтения для таких ситуаций.

from collections import defaultdict 

x = ["A","B","C","D","E"] 
y = [1,2,3,2,1] 
D = defaultdict(list) 

for i, j in zip(x, y): 
    D[j].append(i) 

print dict(D) 

Выход:

{1: ['A', 'E'], 2: ['B', 'D'], 3: ['C']} 

Answer 3

Вот простой (не умный) решение. Я бы сказал, что простое решение больше соответствует философии питонов, чем умному решению. Perl - это язык, призванный максимизировать ум, imho, и я нахожу его почти нечитаемым (по общему признанию, я его избегаю, если я могу быть неопытным программистом Perl).

x = ["A","B","C","D","E"] 
y = [1,2,3,2,1] 

assert(len(x) == len(y)) 

d = {} 
for i in range(len(x)): 
    key = y[i] 
    val = x[i] 

    if key in d: 
     d[key].append(val) 
    else: 
     d[key] = [val, ] 

print d 

Answer 4

Вот умный, но O(n^2) и поэтому не советовал решение, которое я придумал, используя комбинацию словаря & списка понимания Python.

>>> x = ["A","B","C","D","E"] 
>>> y = [1,2,3,2,1] 
>>> {y[i] : [x[j] for j in range(len(y)) if y[j] == y[i]] for i in range(len(y))} 
{1: ['A', 'E'], 2: ['B', 'D'], 3: ['C']} 

За что его ценность, @Joe R и @ mattingly890 решение пути, так как они являются O(n) решения