Как сопоставить ключевые значения с помощью списков?

Question

Я хочу сделать следующее:

1] сравнить каждый элемент в списке a с b и сопоставить аналогичные значения.

a=[1, 2, 3, 6, 4, 5, 7, 8, 9] 
b=[4, 4, 5, 5, 7, 7, 9, 9, 10] 

Я попробовал следующее:

c = set(a) & set(b) 
>>> set([9,4,5,7]) 

2] Я хочу, чтобы эти преобразованные значения в том порядке, нумерации, как появляются в списке Ь, такие как 4,5,7,9.

3] и, наконец, я хочу использовать эти ключевые значения (4,5,7,9) и генерировать соответствующие значения из списка А, например:

4 (1,2) 
5 (3,6) 
7 (4,5) 
9 (7,8) 

Любые предложения будут высоко оценивает?

Answer 1

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

a = [1, 2, 3, 6, 4, 5, 7, 8, 9] 
b = [4, 4, 5, 5, 7, 7, 9, 9, 10] 

r = dict([(x, []) for x in b]) 

for k, v in zip(b, a): 
    r[k] += [v,] 

Вот как для использования результатов:

>>> for k, v in r.items(): 
...  print k, v 
... 
9 [7, 8] 
10 [9,] 
4 [1, 2] 
5 [3, 6] 
7 [4, 5] 
>>> 
>>> print r[4] 
[1, 2] 

Это оптимизированная версия, которая будет работать быстрее с меньшими затратами использование памяти:

from itertools import izip 
from collections import defaultdict 
r = defaultdict(list) 

for k, v in izip(b, a): 
    r[k] += [v] 

Answer 2

Для решения проблемы 3, вам необходимо сравнить каждый лист против противоположного набора. И это решит ваш # 2 автоматически.

Есть и другие способы решения # 2, как с помощью OrderedSet (например, recipe, связанный с collections docs), но так как они не решат № 3, в этом нет смысла.

---

Если a значения очень велики, вы можете создать индекс структуру-а Dict, который отображает значение коллекции индексов-чтобы сделать это быстрее. Как это:

rev_a = collections.defaultdict(list) 
for index, value in enumerate(a): 
    rev_a[value].append(index) 

И тогда поиск часть становится еще проще и быстрее:

for b_value in b: 
    a_indices = rev_a[b_value] 
    if a_indices: 
     a_str = ','.join(map(str, a_indices)) 
     print('{} is associated with ({})'.format(b_value, a_str)) 

Или, если вы хотите создать какую-то структуру вместо печати на лету:

results = [(b_value, rev_a[b_value]) for b_value in b] 
results = OrderedDict((b_value, rev_a[b_value]) for b_value in b) 
# etc. 

---

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

Например, замените a_str линию с этим:

a_str = ','.join(str(a[a_index] for index in a_indices)) 

... и Вы получите желаемый результат.

Answer 3

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

>>> a=[1, 2, 3, 6, 4, 5, 7, 8, 9] 
>>> b=[4, 4, 5, 5, 7, 7, 9, 9, 10] 
>>> [x for x in a if x in b] 
[4, 5, 7, 9]