วิธี Pythonic เพื่อรวมคำสั่ง FOR สำหรับลูปและ IF

ฉันรู้วิธีใช้ทั้งลูปและ if statement ในบรรทัดแยกเช่น:

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

และฉันรู้ว่าฉันสามารถใช้ list comprehension เพื่อรวมสิ่งเหล่านี้เมื่อข้อความง่าย ๆ เช่น:

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

แต่สิ่งที่ฉันไม่สามารถหาได้เป็นตัวอย่างที่ดีที่ใดก็ได้ (เพื่อคัดลอกและเรียนรู้จาก) แสดงชุดคำสั่งที่ซับซ้อน (ไม่ใช่แค่ "พิมพ์ x") ที่เกิดขึ้นหลังจากการรวมกันของ for for loop และบางคำสั่งถ้า สิ่งที่ฉันคาดหวังดูเหมือนว่า:

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

นี่ไม่ใช่วิธีที่งูเหลือมควรจะใช้ใช่ไหม

คุณสามารถใช้นิพจน์ตัวสร้างเช่นนี้:

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

for x in gen:
    print x

ตามThe Zen of Python (หากคุณสงสัยว่ารหัสของคุณเป็น "Pythonic" นั่นคือที่ที่ควรไป):

• สวยดีกว่าน่าเกลียด
• ชัดเจนดีกว่าโดยปริยาย
• เรียบง่ายดีกว่าซับซ้อน
• แบนดีกว่าซ้อนกัน
• จำนวนการอ่าน

วิธี Pythonic ในการรับของทั้งสองคือ:sorted intersectionset

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

หรือองค์ประกอบเหล่านั้นxyzแต่ไม่ได้อยู่ในa:

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

แต่สำหรับลูปที่ซับซ้อนกว่านี้คุณอาจต้องการทำให้แบนโดยการวนซ้ำนิพจน์ตัวกำเนิดที่มีชื่อและ / หรือเรียกใช้ฟังก์ชันที่มีชื่อดี การพยายามที่จะปรับทุกอย่างในบรรทัดเดียวแทบจะไม่ "Pythonic"

อัปเดตการติดตามความคิดเห็นเพิ่มเติมเกี่ยวกับคำถามของคุณและคำตอบที่ยอมรับ

ฉันไม่แน่ใจว่าคุณกำลังพยายามทำอะไรenumerateแต่ถ้าaเป็นพจนานุกรมคุณอาจต้องการใช้ปุ่มเช่นนี้:

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

โดยส่วนตัวฉันคิดว่านี่เป็นเวอร์ชั่นที่สวยที่สุด:

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

แก้ไข

ถ้าคุณกระตือรือร้นที่จะหลีกเลี่ยงการใช้แลมบ์ดาคุณสามารถใช้แอปพลิเคชั่นบางส่วนและใช้โมดูลตัวดำเนินการได้

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

ต่อไปนี้คือการทำให้เข้าใจง่าย / ซับหนึ่งจากคำตอบที่ยอมรับ:

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

แจ้งให้ทราบว่าgeneratorถูกเก็บไว้แบบอินไลน์ นี่คือการทดสอบpython2.7และpython3.6 (สังเกตเห็น parens ในprint;))

ฉันอาจจะใช้:

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

คุณสามารถใช้เครื่องกำเนิดไฟฟ้าได้เช่นกันหากการแสดงออกของเครื่องกำเนิดไฟฟ้ามีส่วนเกี่ยวข้องหรือซับซ้อนเกินไป:

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

for x in gen():
    print x

ใช้intersectionหรือintersection_update

• ทางแยก :

  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)

  จากนั้นbเป็นคำตอบของคุณ

ฉันชอบคำตอบของ Alexเพราะตัวกรองนั้นตรงกับถ้านำไปใช้กับรายการดังนั้นหากคุณต้องการสำรวจส่วนย่อยของรายการที่มีเงื่อนไขนี่น่าจะเป็นวิธีที่เป็นธรรมชาติที่สุด

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)

วิธีนี้มีประโยชน์สำหรับการแยกข้อกังวลหากฟังก์ชั่นเงื่อนไขเปลี่ยนแปลงรหัสเฉพาะซอที่มีคือฟังก์ชันนั้น

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

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

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

กำเนิดวิธีดูเหมือนว่าดีขึ้นเมื่อคุณไม่ต้องการให้สมาชิกของรายการ แต่การปรับเปลี่ยนของสมาชิกกล่าวซึ่งดูเหมือนว่าพอดีมากขึ้นในการให้กำเนิด

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)

นอกจากนี้ตัวกรองยังทำงานร่วมกับเครื่องกำเนิดไฟฟ้าได้ด้วยแม้ว่าในกรณีนี้จะไม่มีประสิทธิภาพ

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)

แต่แน่นอนว่ามันคงจะดีถ้าเขียนแบบนี้:

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)

วิธีง่ายๆในการค้นหาองค์ประกอบทั่วไปที่เป็นเอกลักษณ์ของรายการ a และ b:

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