231

它更多地是关于 python 列表理解语法。我有一个列表理解,可以生成给定范围的奇数列表:

[x for x in range(1, 10) if x % 2]

这构成了一个过滤器 - 我有一个源列表,其中我删除了偶数 ( if x % 2)。我想在这里使用 if-then-else 之类的东西。以下代码失败:

>>> [x for x in range(1, 10) if x % 2 else x * 100]
  File "<stdin>", line 1
    [x for x in range(1, 10) if x % 2 else x * 100]
                                         ^
SyntaxError: invalid syntax

有一个类似 if-else 的 python 表达式:

1 if 0 is 0 else 3

如何在列表理解中使用它?

4

6 回答 6

426

x if y else z是您为每个元素返回的表达式的语法。因此你需要:

[ x if x%2 else x*100 for x in range(1, 10) ]

混淆是因为您在第一个示例中使用了过滤器,但在第二个示例中没有。在第二个示例中,您仅使用三元运算符表达式将每个值映射到另一个值。

使用过滤器,您需要:

[ EXP for x in seq if COND ]

如果没有过滤器,您需要:

[ EXP for x in seq ]

在您的第二个示例中,表达式是一个“复杂”的表达式,恰好涉及if-else.

于 2013-06-26T13:18:58.253 回答
25
[x if x % 2 else x * 100 for x in range(1, 10) ]
于 2013-06-26T13:19:43.530 回答
16

您也可以通过列表理解来做到这一点:

A=[[x*100, x][x % 2 != 0] for x in range(1,11)]
print A
于 2014-06-29T07:40:36.000 回答
13

只是另一种解决方案,希望有人会喜欢它:

使用:[假,真] [表达式]

>>> map(lambda x: [x*100, x][x % 2 != 0], range(1,10))
[1, 200, 3, 400, 5, 600, 7, 800, 9]
>>>
于 2014-01-18T12:54:12.043 回答
2

#如何排列一个负值和正值数组的列表

my_list=[-2,-3,0,1,5]
squred_values=[]
  

squred_values=[-i**2 if i<0 else i**2 for i in my_list]

#或者

for i in my_list:
    if i<0:
        squred_values.append( -i**2)`enter code here`
    else:
        squred_values.append( i**2) 
于 2021-09-09T20:39:17.317 回答
0

二合一列表理解

ex = [['obi', 'is', '#alive'],['oge', 'is', 'beautiful'],
                ['Ade', 'the', '#comedian', 'de', '#rich'],['Jesus', 'wept']]
res = [j if j.startswith("#") else "_"+j for i in ex for j in i]
于 2021-12-02T15:56:18.640 回答