Sortie inattendue lors du filtrage de la liste Python: Que fais-je de mal?

Question

Je suis en train de filtrer une liste, voici le code:

test=['aaa','bbb','ccc','ddd','footer','header'] 

def rm_hf(x): return x != 'footer' 

filter(rm_hf,test) 

Résultats dans:

>>>['aaa','bbb','ccc','ddd','header'] 

C'est le résultat attendu, trouver « pied » dans la liste et le retirer.

Maintenant, je veux supprimer à la fois « en-tête » et « bas de page », donc je le fais:

test2=['aaa','bbb','ccc','ddd','footer','header'] 

def rm_hf2(x): return x != 'footer' or x != 'header' 

filter(rm_hf2,test2) 

Résultats dans:

>>>['aaa','bbb','ccc','ddd','footer','header'] 

Maintenant qui est bizarre, il donne juste à la fois « pied de page ',' header 'au lieu de les filtrer?

Qu'est-ce que j'ai fait de mal? Je pense que ma logique est correcte ...

Answer 1

Votre logique est correcte parce que vous pensez comme un humain. Votre ordinateur ne le fait pas. Il lit chaque élément de votre liste, puis trébuche sur «pied de page». "Le footer est-il différent du footer?", Dit-il. "NON, c'est la même chaîne, elle est évaluée à false, voyons la condition suivante". "Le pied de page est-il différent de l'en-tête?" La condition est donc False or True, ce qui évalue évidemment à vrai.

Vous voulez un and, pas or:

def rm_hf2(x): return x != 'footer' and x != 'header' 

Vous pouvez également utiliser un tuple et le mot-clé in, qui est plus lisible:

def rm_hf2(x): return x not in ('footer', 'header') 

Il est important que vous compreniez ce qui est vraiment continuer avec "et" et "ou", cependant. Et soyons honnêtes: si quelque chose ne fonctionne pas comme vous le pensez, le problème réside probablement dans votre propre code, et pas dans le langage Python lui-même.

Answer 2

Ce que tout le monde dit d'autre, plus:

Lorsque vous avez plusieurs éléments que vous voulez exclure, utilisez un set au lieu d'une chaîne de and s ou tuple:

# do once 
blacklist = set(['header', 'footer']) 

# as needed 
filter(lambda x: x not in blacklist, some_iterable) 

Justification: Recherche tuple prend un temps proportionnel à la position de l'élément trouvé; l'échec prend le même temps que le dernier élément. Rechercher un élément dans un ensemble prend le même temps pour tous les éléments, et pour l'échec. Les ensembles gagnent généralement pour un grand nombre d'objets. Tout dépend de la probabilité que chaque élément soit recherché et de la probabilité d'échec. Les tuples peuvent gagner même avec une grande collection quand il y a une forte probabilité de quelques objets (ils devraient être placés à l'avant de l'uplet) et un faible risque d'échec.

Answer 3

Vous pouvez également utiliser une liste de compréhension à la place du filtre.

test = ['aaa','bbb','ccc','ddd','footer','header'] 
filtered_test = [x for x in test if x not in ('footer', 'header')] 

ou une expression du générateur (en fonction de vos besoins)

test = ['aaa','bbb','ccc','ddd','footer','header'] 
filtered_test = (x for x in test if x not in ('footer', 'header')) 

Answer 4

my logic is correct

En fait, non, ce n'est pas, comme l'a souligné dans d'autres réponses.

loin plus propre moyen d'atteindre le résultat souhaité est d'utiliser list comprehensions, savoir:

test = ['aaa', 'bbb', 'ccc', 'ddd', 'footer', 'header'] 
undesirable = ['footer', 'header'] 
[_ for _ in test if _ not in undesirable] 

De the documentation:

Note that filter(function, iterable) is equivalent to [item for item in iterable if function(item)] if function is not None and [item for item in iterable if item] if function is None .

Cela dit, il n'y a pas de temps comme le présent pour brosser sur votre Boolean logic!

Si vous deviez tester votre code à l'unité, vous découvrirez rapidement que votre deuxième fonction de filtrage ne répond pas à vos attentes. Voici un exemple simpliste:

$ cat 4281875.py 
#!/usr/bin/env python 

import unittest 

def rm_hf2(x): return x != 'footer' or x != 'header' 

class test_rm_hft(unittest.TestCase): 

    def test_aaa_is_not_filtered(self): 
     self.assertTrue(rm_hf2('aaa')) 

    def test_footer_is_filtered_out(self): 
     self.assertFalse(rm_hf2('footer')) 


if __name__ == '__main__': 
    unittest.main() 


$ ./4281875.py 
.F 
====================================================================== 
FAIL: test_footer_is_filtered_out (__main__.test_rm_hft) 
---------------------------------------------------------------------- 
Traceback (most recent call last): 
    File "./4281875.py", line 13, in test_footer_is_filtered_out 
    self.assertFalse(rm_hf2('footer')) 
AssertionError 

---------------------------------------------------------------------- 
Ran 2 tests in 0.000s 

FAILED (failures=1)