У меня есть список 'a'
a= [(1,2),(1,4),(3,5),(5,7)]
Мне нужно найти все кортежи для определенного числа. скажем, за 1 будет
result = [(1,2),(1,4)]
Как это сделать?
У меня есть список '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
Использование функции фильтра:
>>> def get_values(iterables, key_to_find):
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)]
>>> [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))
Или 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)]
>>>