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

Question

У меня есть список 'a'

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

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

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

Как мне это сделать?

Answer 1

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

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

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

[item for item in a if 1 in item]

Answer 2

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

Например,

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

Answer 3

Читать дальше Понимание списка

[ (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 ) )

Answer 4

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

Answer 5

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

Answer 6

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

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

Answer 7

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

>>> def get_values(iterables, key_to_find):<br/>
        return list(filter(lambda x:key_to_find in x, iterables)) 

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

>>> get_values(a, 1)

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

Answer 8

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

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], если хотите найти номер по определенному индексу

Answer 9

Или 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)]
>>> 

Answer 10

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

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

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

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