Найти совпадающие значения в списке списков с помощью python

Question

Я пытаюсь перебрать список списков в python 2.7.5 и вернуть те, где первое значение найдено во втором списке, примерно так:

#python 2.7.5 
list1 = ['aa', 'ab', 'bb', 'bc', 'cc'] 
list2 = [['aa', 1, 3, 7],['de', 2, 2, 1],['bc', 3, 4, 4]] 

list3 = [] 
for x in list1: 
    for y in list2: 
     if x == y: 
      list3.append(y) 

Поэтому я бы хотел, чтобы list3 содержал [['aa',1,3,7],['bc', 3, 4, 4]], но вместо этого я просто получил весь список2.

Answer 1

Попробуйте более простой подход, который ближе к тому, что вы хотите:

for e in list2: 
    if e[0] in list1: 
     list3.append(e) 

Вам нужно e[0] так list2 список списков. Вы также можете написать в одной строке с помощью filter() function:

list3 = filter(lambda e: e[0] in list1, list2) 

или с помощью list comprehension:

list3 = [e for e in list2 if e[0] in list1] 

Answer 2

В коде есть только ошибка. Вам нужно x == y [0] вместо x == y. Последний сравнивает строку со списком.

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

>>> [x for x in list2 if x[0] in list1] 
[['aa', 1, 3, 7],['bc', 3, 4, 4]] 

Answer 3

Похоже, ваш вопрос, потому что вы пытаетесь сравнить список элементов со списком списка элементов.

Если вы шагаете через петлю for, вы увидите, что на вашей первой итерации вы сравниваете 'aa' против ['aa', 1, 3, 7], который, вероятно, не то, что вы хотели бы сделать.

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

for x in list1: 
    for y in list2: 
     if x == y[0]: 
      list3.append(y)