Cara Pythonic untuk menggabungkan pernyataan FOR loop dan IF

Saya tahu cara menggunakan keduanya untuk loop dan jika pernyataan pada baris terpisah, seperti:

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

Dan saya tahu saya bisa menggunakan daftar pemahaman untuk menggabungkan ini ketika pernyataan sederhana, seperti:

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

Tetapi apa yang saya tidak dapat temukan adalah contoh yang baik di mana saja (untuk menyalin dan belajar dari) menunjukkan serangkaian perintah yang kompleks (bukan hanya "cetak x") yang terjadi setelah kombinasi dari loop for dan beberapa pernyataan if. Sesuatu yang saya harapkan terlihat seperti:

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

Apakah ini bukan cara python seharusnya bekerja?

Anda dapat menggunakan ekspresi generator seperti ini:

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

for x in gen:
    print x

Per The Zen of Python (jika Anda bertanya-tanya apakah kode Anda adalah "Pythonic", itu adalah tempat untuk pergi):

• Cantik lebih baik dari yang jelek.
• Eksplisit lebih baik daripada implisit.
• Sederhana lebih baik daripada kompleks.
• Flat lebih baik daripada bersarang.
• Jumlah keterbacaan diperhitungkan.

Cara Pythonic untuk mendapatkan keduanya adalah:sorted intersectionset

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

Atau elemen-elemen yang ada xyztetapi tidak di a:

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

Tetapi untuk loop yang lebih rumit, Anda mungkin ingin meratakannya dengan mengulangi ekspresi generator yang disebut dan / atau memanggil ke fungsi yang telah ditentukan. Mencoba menyesuaikan segala sesuatunya dalam satu baris jarang "Pythonic".

Perbarui komentar tambahan berikut pada pertanyaan Anda dan jawaban yang diterima

Saya tidak yakin apa yang ingin Anda lakukan enumerate, tetapi jika aini adalah kamus, Anda mungkin ingin menggunakan kunci, seperti ini:

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

Saya pribadi berpikir ini adalah versi tercantik:

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

Edit

jika Anda sangat ingin menghindari penggunaan lambda, Anda dapat menggunakan aplikasi fungsi parsial dan menggunakan modul operator (yang menyediakan fungsi sebagian besar operator).

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

Berikut ini adalah penyederhanaan / satu kalimat dari jawaban yang diterima:

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

Perhatikan bahwa generatoritu tetap sejajar . Ini diuji pada python2.7dan python3.6 (perhatikan parens di print;))

Saya mungkin akan menggunakan:

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

Anda dapat menggunakan generator juga, jika ekspresi generator menjadi terlalu terlibat atau kompleks:

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

for x in gen():
    print x

Gunakan intersectionatauintersection_update

• persimpangan :

  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)

  maka bjawaban Anda adalah

Saya menyukai jawaban Alex , karena filter adalah persis jika diterapkan ke daftar, jadi jika Anda ingin menjelajahi subset daftar yang diberikan syarat, ini sepertinya cara yang paling alami

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)

metode ini berguna untuk pemisahan masalah, jika fungsi kondisinya berubah, satu-satunya kode yang digunakan adalah fungsi itu sendiri

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

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

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

The Generator Metode tampaknya lebih baik bila Anda tidak ingin anggota daftar, tapi modifikasi dari anggota mengatakan, yang tampaknya lebih cocok untuk sebuah pembangkit

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)

Juga, filter bekerja dengan generator, meskipun dalam kasus ini tidak efisien

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)

Tapi tentu saja, tetap menyenangkan menulis seperti ini:

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)

Cara sederhana untuk menemukan elemen umum yang unik dari daftar a dan b:

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