Блог

Главная - Вопросы по программированию - Извлечь подсписок из списка в Python

Извлечь подсписок из списка в Python

#python #list

#python #Список

Вопрос:

как я могу извлечь, например, [[‘A’, ‘123’], [‘A’, ‘456’]] из моего списка, если я отфильтровал по ‘A’?

 mylist = [['A', '123'],
 ['A', '456'],
 ['B','847'],
 ['B','677']]

Ответ №1:

Вот два способа достижения желаемых результатов.

 mylist = [['A', '123'],
 ['A', '456'],
 ['B','847'],
 ['B','677']]
letter = 'A'
# Using list comprehension
print([l for l in mylist if l[0] == letter])
# Using filer function 
print(list(filter(lambda l: l[0] == letter, mylist)))

Ответ №2:

Я создал код для вас.

 mylist = [['A', '123'],
          ['A', '456'],
          ['B', '847'],
          ['B', '677']]

output = [lst for lst in mylist if 'A' in lst]
print(output)

Или вы можете использовать этот код;

 output = [lst for lst in mylist if 'A' == lst[0]]