Найти элемент в списке кортежей


140

У меня есть список «а»

a= [(1,2),(1,4),(3,5),(5,7)]

Мне нужно найти все кортежи для определенного номера. скажем за 1 это будет

result = [(1,2),(1,4)]

Как я могу это сделать?

Ответы:


236

Если вы просто хотите, чтобы первое число совпадало, вы можете сделать это так:

[item for item in a if item[0] == 1]

Если вы просто ищете кортежи с 1 в них:

[item for item in a if 1 in item]

115

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

Например,

test = [("hi", 1), ("there", 2)]
test = dict(test)
print test["hi"] # prints 1

11
Как вы примените это к проблеме Брюса?
Здравствуйте, до свидания

5
Хороший ответ (хотя, возможно, не для этого вопроса). Мне удалось определить, есть ли значение в списке выбранных кортежей (например, если в тесте «привет»)
MagicLAMP,

10
Это действительно не отвечает на вопрос, как предлагает MagicLAMP. В частности, dict(X)преобразует X в словарь, где последний кортеж любого общего первого элемента - это используемое значение. В примере OP это вернуло бы (1,4) в отличие от (1,2) и (1,4).
BBischof





1

filterФункция может также предоставить интересное решение:

result = list(filter(lambda x: x.count(1) > 0, a))

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

result = list(filter(lambda x: x[0] == 1, a))

1

Используя функцию фильтра:

>>> def get_values ​​(iterables, key_to_find):
возвращаемый список (фильтр (лямбда-x: key_to_find в x, итерируемые)) >>> a = [(1,2), (1,4), (3,5), (5,7)] >>> get_values ​​(a, 1) >>> [(1, 2), (1, 4)]

1

Или takewhile(в дополнение к этому показан пример большего количества значений):

>>> a= [(1,2),(1,4),(3,5),(5,7),(0,2)]
>>> import itertools
>>> list(itertools.takewhile(lambda x: x[0]==1,a))
[(1, 2), (1, 4)]
>>> 

если не отсортировано, например:

>>> a= [(1,2),(3,5),(1,4),(5,7)]
>>> import itertools
>>> list(itertools.takewhile(lambda x: x[0]==1,sorted(a,key=lambda x: x[0]==1)))
[(1, 2), (1, 4)]
>>> 

0

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

a= [(1,2),(1,4),(3,5),(5,7)]
i=1
result=[]
for j in a:
    if i in j:
        result.append(j)

print(result)

Вы также можете использовать, if i==j[0] or i==j[index]если вы хотите найти номер в определенном индексе

Используя наш сайт, вы подтверждаете, что прочитали и поняли нашу Политику в отношении файлов cookie и Политику конфиденциальности.
Licensed under cc by-sa 3.0 with attribution required.