Trouver un élément dans une liste de tuples
J'ai une liste 'a'
a= [(1,2),(1,4),(3,5),(5,7)]
J'ai besoin de trouver toutes les tuples pour un nombre particulier. dites pour 1 ce sera
result = [(1,2),(1,4)]
Comment puis-je faire ça?
8 réponses
Si vous voulez juste que le premier nombre corresponde, vous pouvez le faire comme ceci:
[item for item in a if item[0] == 1]
Si vous recherchez simplement des tuples avec 1 en eux:
[item for item in a if 1 in item]
Il existe en fait une façon intelligente de le faire qui est utile pour toute liste de tuples où la taille de chaque tuple est 2: Vous pouvez convertir votre liste en un seul dictionnaire.
Par exemple
test = [("hi", 1), ("there", 2)]
test = dict(test)
print test["hi"] # prints 1
Lire sur Liste compréhensions
[ (x,y) for x, y in a if x == 1 ]
Également lire jusqu' générateur de fonctions et le yield
déclaration.
def filter_value( someList, value ):
for x, y in someList:
if x == value :
yield x,y
result= list( filter_value( a, 1 ) )
La fonction filter
peut également fournir une solution intéressante:
result = list(filter(lambda x: x.count(1) > 0, a))
Qui recherche les tuples dans la liste pour toute occurrence de 1
. Si la recherche est limitée au premier élément, la solution peut être modifié en:
result = list(filter(lambda x: x[0] == 1, a))
Utilisation de la fonction de filtre:
>>> 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)]