结合 for 循环和 if 语句的 Pythonic 方式

Question

我知道如何在单独的行上使用 for 循环和 if 语句，例如：

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

我知道当语句很简单时，我可以使用列表理解来组合这些语句，例如：

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

但我找不到一个很好的例子任何地方（复制和学习）演示一组复杂的命令（不仅仅是“print x”），这些命令发生在 for 循环和一些 if 语句的组合之后。我期望的东西看起来像：

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

这不是 python 应该工作的方式吗？

Answer 1

如果生成器表达式变得过于复杂或复杂，您也可以使用 生成器：

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

for x in gen():
    print x

Answer 2

基于此处的文章：https://towardsdatascience.com/a-compressive-hands-on-guide-to-transfer-learning-with-real-world-applications-in-deep-learning-212bf3b2f27a
出于同样的原因，我使用了以下代码，并且运行得很好：

an_array = [x for x in xyz if x not in a]

这一行是程序的一部分！这意味着 XYZ 是一个要预先定义和分配的数组，也是变量 a

使用生成器表达式（在所选答案中推荐）会产生一些困难，因为结果不是一个数组

Answer 3

使用 intersection 或 intersection_update

• intersection ：

  <前><代码>a = [2,3,4,5,6,7,8,9,0]
  xyz = [0,12,4,6,242,7,9]
  ans = 排序(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 = 集合(a)
  b.intersection_update(xyz)

  那么 b 就是你的答案

Answer 4

嗯，只需一行即可完成。

a = [2,3,4,5,6,7,8,9,0]
xyz = [0,12,4,6,242,7,9]
print('\n'.join([str(x) for x in xyz if x not in a]))
-----------------------------------------------------------------------
12
242

This gives you:

Answer 5

虽然我真的很喜欢已接受的答案，但我想补充

for x in xyz:
    if x not in a:
        continue  # jump to the next element of xyz
    print(x)

一下讨论。

“优势”是有争议的，但正如这里已经指出的，并引用自 The Zen of Python，“简单胜于复杂”和“可读性很重要”。我声称 continue 的概念没有生成器表达式那么复杂。

Answer 6

查找列表 a 和 b 的唯一公共元素的简单方法：

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

Answer 7

您可以使用 生成器表达式，如下所示：

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

for x in gen:
    print(x)

Answer 8

根据The Zen of Python（如果你想知道你的代码是否是“Pythonic”，那就是去的地方）：

• 美丽胜于丑陋。
• 显式的比隐式的好。
• 简单总比复杂好。
• 扁平比嵌套更好。
• 可读性很重要。

获取 sorted 的 Pythonic 方式 两个 href="http://www.python.org/doc//current/library/stdtypes.html#set">set 是：

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

Answer 9

以下是已接受答案的简化/一行：

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

请注意，生成器保持内联。这是在 python2.7 和 python3.6 上进行了测试（注意 print 中的括号；））

即使如此，老实说还是很麻烦： x 被提及四次。

Answer 10

我个人认为这是最漂亮的版本：

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

编辑

如果您非常热衷于避免使用 lambda，您可以使用部分函数应用程序并使用运算符模块（提供大多数运算符的功能）。

https://docs.python.org/2/library/operator.html #模块运算符

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

Answer 11

我可能会使用：

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

Answer 12

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

Answer 13

我喜欢 Alex 的回答，因为filter 正是应用于列表的 if，因此，如果您想在给定条件下探索列表的子集，这似乎是最有效的自然地，

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)