Знайдіть елемент у списку кортежів

У мене є список "a"

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

Мені потрібно знайти всі кортежі для певної кількості. скажіть, для 1 це буде

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

Як це зробити?

Якщо ви просто хочете відповідати першому номеру, ви можете зробити це так:

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

Якщо ви просто шукаєте кортежі з 1 в них:

[item for item in a if 1 in item]

Насправді є розумний спосіб зробити це корисним для будь-якого списку кортежів, де розмір кожного кортежу становить 2: ви можете перетворити свій список в один словник.

Наприклад,

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

Прочитайте на Зрозуміння списку

[ (x,y) for x, y in a if x  == 1 ]

Прочитайте також функції генератора та yieldзаяву.

def filter_value( someList, value ):
    for x, y in someList:
        if x == value :
            yield x,y

result= list( filter_value( a, 1 ) )

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

for item in a:
   if 1 in item:
       print item

>>> [i for i in a if 1 in i]

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

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

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

який шукає кортежі в списку для будь-якого виникнення 1. Якщо пошук обмежений першим елементом, рішення можна змінити на:

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

Використання функції фільтра:

>>> визначити get_values ​​(ітерабелі, 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)]

Або 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)]
>>> 

якщо ви хочете шукати в кортежі будь-яке число, яке є в кортежі, тоді ви можете використовувати

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]якщо ви хочете шукати номер у конкретному індексі