Znajdź element na liście krotek

140

Mam listę 'a'

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

Muszę znaleźć wszystkie krotki dla określonej liczby. powiedz za 1 to będzie

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

W jaki sposób mogę to zrobić?

Bruce
źródło

Odpowiedzi:

236

Jeśli chcesz tylko dopasować pierwszą liczbę, możesz to zrobić w ten sposób:

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

Jeśli szukasz tylko krotek z 1 w nich:

[item for item in a if 1 in item]
Nadia Alramli
źródło
115

W rzeczywistości istnieje sprytny sposób, aby to zrobić, który jest przydatny w przypadku każdej listy krotek, w których rozmiar każdej krotki wynosi 2: możesz przekonwertować swoją listę na pojedynczy słownik.

Na przykład,

test = [("hi", 1), ("there", 2)]
test = dict(test)
print test["hi"] # prints 1
Steven Holtzen
źródło
11
Jak odnosisz to do problemu Bruce'a?
HelloGoodbye
5
Dobra odpowiedź (choć prawdopodobnie nie na to pytanie). Pomogło mi ustalić, czy wartość znajduje się na liście krotek do wyboru (np. Jeśli w teście „cześć”)
MagicLAMP,
10
Tak naprawdę nie odpowiada na pytanie, jak sugeruje MagicLAMP. W szczególności dict(X)konwertuje X na słownik, w którym ostatnia krotka dowolnego wspólnego pierwszego elementu jest używaną wartością. W przykładzie PO zwróciłoby to (1,4), w przeciwieństwie do (1,2) i (1,4).
BBischof
23

Przeczytaj więcej na temat rozumienia list

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

Przeczytaj także funkcje generatora i yieldinstrukcję.

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

result= list( filter_value( a, 1 ) )
S.Lott
źródło
1
if x == 1powinno byćif x == value
sambha
10
[tup for tup in a if tup[0] == 1]
Tendayi Mawushe
źródło
8
for item in a:
   if 1 in item:
       print item
ghostdog74
źródło
2
>>> [i for i in a if 1 in i]

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

suyash
źródło
1

filterFunkcja może również interesujące rozwiązanie:

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

który przeszukuje krotki na liście pod kątem wystąpienia 1. Jeśli wyszukiwanie jest ograniczone do pierwszego elementu, rozwiązanie można zmodyfikować na:

result = list(filter(lambda x: x[0] == 1, a))
MxNx
źródło
1

Korzystanie z funkcji filtra:

>>> def get_values ​​(iterable, key_to_find):
lista zwrotów (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)]
pradam
źródło
1

Lub takewhile(oprócz tego pokazano przykład większej liczby wartości):

>>> 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)]
>>> 

jeśli nieposortowane, na przykład:

>>> 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)]
>>> 
U10-Forward
źródło
0

jeśli chcesz wyszukać w krotce dowolną liczbę obecną w krotce, możesz użyć

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)

Możesz również użyć, if i==j[0] or i==j[index]jeśli chcesz wyszukać numer w określonym indeksie

sunil nagori
źródło