Cách Pythonic để kết hợp vòng lặp FOR và câu lệnh IF

Tôi biết cách sử dụng cả hai cho các vòng lặp và nếu các câu lệnh trên các dòng riêng biệt, chẳng hạn như:

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

Và tôi biết rằng tôi có thể sử dụng một sự hiểu biết danh sách để kết hợp những điều này khi các câu lệnh đơn giản, chẳng hạn như:

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

Nhưng những gì tôi không thể tìm thấy là một ví dụ tốt ở bất cứ đâu (để sao chép và học hỏi) thể hiện một tập lệnh phức tạp (không chỉ là "in x") xảy ra sau sự kết hợp của vòng lặp for và một số câu lệnh if. Một cái gì đó mà tôi mong đợi trông giống như:

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

Đây có phải chỉ là cách con trăn được cho là hoạt động?

Bạn có thể sử dụng các biểu thức trình tạo như thế này:

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

for x in gen:
    print x

Theo Zen của Python (nếu bạn đang tự hỏi liệu mã của mình có phải là "Pythonic" hay không, đó là nơi cần đến):

• Đẹp thì tốt hơn xấu.
• Rõ ràng là tốt hơn so với ngầm.
• Đơn giản là tốt hơn phức tạp.
• Bằng phẳng là tốt hơn so với lồng nhau.
• Tính dễ đọc.

Cách Pythonic để có được hai s là:sorted intersectionset

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

Hoặc những yếu tố đó xyznhưng không có trong a:

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

Nhưng đối với một vòng lặp phức tạp hơn, bạn có thể muốn làm phẳng nó bằng cách lặp qua biểu thức trình tạo có tên tốt và / hoặc gọi ra một hàm có tên tốt. Cố gắng để phù hợp với tất cả mọi thứ trên một dòng hiếm khi là "Pythonic".

Cập nhật sau nhận xét bổ sung về câu hỏi của bạn và câu trả lời được chấp nhận

Tôi không chắc chắn những gì bạn đang cố gắng làm với enumerate, nhưng nếu alà một từ điển, bạn có thể muốn sử dụng các phím, như thế này:

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

Cá nhân tôi nghĩ rằng đây là phiên bản đẹp nhất:

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

Biên tập

nếu bạn rất muốn tránh sử dụng lambda, bạn có thể sử dụng ứng dụng chức năng một phần và sử dụng mô đun toán tử (cung cấp chức năng của hầu hết các toán tử).

https://docs.python.org/2/l Library / operator.html # model-operator

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

Sau đây là đơn giản hóa / một lớp lót từ câu trả lời được chấp nhận:

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

Lưu ý rằng generatorđã được giữ nội tuyến . Điều này đã được thử nghiệm trên python2.7và python3.6 (chú ý các dấu ngoặc trong print;))

Tôi có thể sẽ sử dụng:

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

Bạn cũng có thể sử dụng trình tạo , nếu các biểu thức của trình tạo trở nên quá liên quan hoặc phức tạp:

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

for x in gen():
    print x

Sử dụng intersectionhoặcintersection_update

• ngã tư :

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

• giao lộ_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)

  sau đó blà câu trả lời của bạn

Tôi thích câu trả lời của Alex , vì bộ lọc chính xác là nếu được áp dụng cho danh sách, vì vậy nếu bạn muốn khám phá một tập hợp con của danh sách được cung cấp một điều kiện, đây có vẻ là cách tự nhiên nhất

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)

phương pháp này hữu ích cho việc phân tách các mối quan tâm, nếu hàm điều kiện thay đổi, mã duy nhất để tìm hiểu là chính hàm đó

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

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

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

Các máy phát điện phương pháp có vẻ tốt hơn khi bạn không muốn các thành viên của danh sách, nhưng sự thay đổi của các thành viên cho biết, mà dường như nhiều hơn phù hợp với một máy phát điện

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)

Ngoài ra, các bộ lọc hoạt động với máy phát điện, mặc dù trong trường hợp này nó không hiệu quả

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)

Nhưng tất nhiên, vẫn sẽ rất tuyệt nếu viết như thế này:

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)

Một cách đơn giản để tìm các yếu tố phổ biến duy nhất của danh sách a và b:

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