列表理解与条件

时间:2014-06-26 23:59:02

标签: python if-statement list-comprehension generator-expression

我有一个简单的清单。

>>> a = [0, 1, 2]

我想使用列表推导从中创建一个新列表。

>>> b = [x*2 for x in a]
>>> b
[0, 2, 4]

非常简单,但如果我只想操作非零元素呢?

'如果'在列表推导中需要'其他',那么我想出了这个。

>>> b = [x*2 if x != 0 else None for x in a]
>>> b
[None, 2, 4]

但理想的结果是。

>>> b
[2, 4]

我可以这样做

>>> a = [0, 1, 2]
>>> def f(arg):
...     for x in arg:
...         if x != 0:
...             yield x*2
... 
>>> list(f(a))
[2, 4]

或使用'filter'和lambda

>>> a = [0, 1, 2]
>>> list(filter(lambda x: x != 0, a))
[1, 2]

如何使用列表推导获得此结果?

2 个答案:

答案 0 :(得分:17)

b = [x*2 for x in a if x != 0]

如果你把条件放在最后,你不需要别的(事实上不能有其他的)

答案 1 :(得分:7)

遵循模式:

[ <item_expression>
  for <item_variables> in <iterator>
  if <filtering_condition>
]

我们可以解决它:

>>> lst = [0, 1, 2]
>>> [num
... for num in lst
... if num != 0]
[1, 2]

所有关于形成if条件测试“非零”值。