Pythoniczny sposób łączenia pętli FOR i instrukcji IF

Wiem, jak używać zarówno do pętli, jak i instrukcji w osobnych wierszach, takich jak:

>>> a = [2,3,4,5,6,7,8,9,0]
... xyz = [0,12,4,6,242,7,9]
... for x in xyz:
...     if x in a:
...         print(x)
0,4,6,7,9

I wiem, że mogę użyć rozumienia listy, aby połączyć je, gdy stwierdzenia są proste, takie jak:

print([x for x in xyz if x in a])

Ale nie mogę znaleźć nigdzie dobrego przykładu (do skopiowania i uczenia się) demonstrującego złożony zestaw poleceń (nie tylko „print x”), które występują po kombinacji pętli for i niektórych instrukcji if. Coś, czego oczekiwałbym, wygląda:

for x in xyz if x not in a:
    print(x...)

Czy to nie tak powinien działać Python?

Możesz użyć wyrażeń generatora takich jak to:

gen = (x for x in xyz if x not in a)

for x in gen:
    print x

Zgodnie z Zen of Python (jeśli zastanawiasz się, czy Twój kod to „ Python ”, to jest miejsce, do którego należy przejść):

• Piękne jest lepsze niż brzydkie.
• Jawne jest lepsze niż niejawne.
• Prosty jest lepszy niż złożony.
• Mieszkanie jest lepsze niż zagnieżdżone.
• Liczy się czytelność.

Pythoniczny sposób uzyskania dwóch s jest następujący:sorted intersectionset

>>> sorted(set(a).intersection(xyz))
[0, 4, 6, 7, 9]

Lub te elementy, które xyznie są w a:

>>> sorted(set(xyz).difference(a))
[12, 242]

Ale w przypadku bardziej skomplikowanej pętli możesz ją spłaszczyć, wykonując iterację nad dobrze nazwanym wyrażeniem generatora i / lub wywołując dobrze nazwaną funkcję. Próbowanie dopasowania wszystkiego w jednym wierszu rzadko jest „pytoniczne”.

Zaktualizuj po dodatkowych komentarzach do pytania i zaakceptowanej odpowiedzi

Nie jestem pewien, co próbujesz zrobić enumerate, ale jeśli ajest słownikiem, prawdopodobnie chcesz użyć klawiszy:

>>> a = {
...     2: 'Turtle Doves',
...     3: 'French Hens',
...     4: 'Colly Birds',
...     5: 'Gold Rings',
...     6: 'Geese-a-Laying',
...     7: 'Swans-a-Swimming',
...     8: 'Maids-a-Milking',
...     9: 'Ladies Dancing',
...     0: 'Camel Books',
... }
>>>
>>> xyz = [0, 12, 4, 6, 242, 7, 9]
>>>
>>> known_things = sorted(set(a.iterkeys()).intersection(xyz))
>>> unknown_things = sorted(set(xyz).difference(a.iterkeys()))
>>>
>>> for thing in known_things:
...     print 'I know about', a[thing]
...
I know about Camel Books
I know about Colly Birds
I know about Geese-a-Laying
I know about Swans-a-Swimming
I know about Ladies Dancing
>>> print '...but...'
...but...
>>>
>>> for thing in unknown_things:
...     print "I don't know what happened on the {0}th day of Christmas".format(thing)
...
I don't know what happened on the 12th day of Christmas
I don't know what happened on the 242th day of Christmas

Osobiście uważam, że jest to najładniejsza wersja:

a = [2,3,4,5,6,7,8,9,0]
xyz = [0,12,4,6,242,7,9]
for x in filter(lambda w: w in a, xyz):
  print x

Edytować

jeśli bardzo zależy ci na uniknięciu użycia lambda, możesz użyć częściowej aplikacji funkcji i użyć modułu operatora (który zapewnia funkcje większości operatorów).

https://docs.python.org/2/library/operator.html#module-operator

from operator import contains
from functools import partial
print(list(filter(partial(contains, a), xyz)))

Poniżej znajduje się uproszczenie / jedna linijka z zaakceptowanej odpowiedzi:

a = [2,3,4,5,6,7,8,9,0]
xyz = [0,12,4,6,242,7,9]

for x in (x for x in xyz if x not in a):
    print(x)

12
242

Zwróć uwagę, że generatorzostał utrzymany w linii . Zostało to przetestowane python2.7i python3.6 (zauważ pareny w print;))

Prawdopodobnie użyłbym:

for x in xyz: 
    if x not in a:
        print x...

a = [2,3,4,5,6,7,8,9,0]
xyz = [0,12,4,6,242,7,9]  
set(a) & set(xyz)  
set([0, 9, 4, 6, 7])

Możesz także użyć generatorów , jeśli wyrażenia generatorów stają się zbyt skomplikowane lub złożone:

def gen():
    for x in xyz:
        if x in a:
            yield x

for x in gen():
    print x

Użyj intersectionlubintersection_update

• skrzyżowanie :

  a = [2,3,4,5,6,7,8,9,0]
  xyz = [0,12,4,6,242,7,9]
  ans = sorted(set(a).intersection(set(xyz)))

• intersection_update :

  a = [2,3,4,5,6,7,8,9,0]
  xyz = [0,12,4,6,242,7,9]
  b = set(a)
  b.intersection_update(xyz)

  to bjest twoja odpowiedź

Lubiłem odpowiedź Alexa , ponieważ filtr jest akuratnie jeśli stosuje się do listy, więc jeśli chcesz poznać podzbiór listy dany warunek, to wydaje się być najbardziej naturalnym sposobem

mylist = [1,2,3,4,5]
another_list = [2,3,4]

wanted = lambda x:x in another_list

for x in filter(wanted, mylist):
    print(x)

metoda ta jest przydatna do rozdzielenia problemów, jeśli zmieni się funkcja warunku, jedynym kodem do manipulowania jest sama funkcja

mylist = [1,2,3,4,5]

wanted = lambda x:(x**0.5) > 10**0.3

for x in filter(wanted, mylist):
    print(x)

Metoda generatora wydaje się lepsza, gdy nie chcesz członków listy, ale modyfikacja tych elementów, która wydaje się bardziej dopasowana do generatora

mylist = [1,2,3,4,5]

wanted = lambda x:(x**0.5) > 10**0.3

generator = (x**0.5 for x in mylist if wanted(x))

for x in generator:
    print(x)

Ponadto filtry działają z generatorami, chociaż w tym przypadku nie jest to wydajne

mylist = [1,2,3,4,5]

wanted = lambda x:(x**0.5) > 10**0.3

generator = (x**0.9 for x in mylist)

for x in filter(wanted, generator):
    print(x)

Ale oczywiście byłoby miło napisać w ten sposób:

mylist = [1,2,3,4,5]

wanted = lambda x:(x**0.5) > 10**0.3

# for x in filter(wanted, mylist):
for x in mylist if wanted(x):
    print(x)

Prosty sposób na znalezienie unikalnych wspólnych elementów list a i b:

a = [1,2,3]
b = [3,6,2]
for both in set(a) & set(b):
    print(both)