单行列表理解:if-else变体
这是关于Python列表理解语法。 我有一个列表理解产生给定范围的奇数列表:
[x for x in range(1, 10) if x % 2]
这使得一个filter – 我有一个源列表,我删除偶数( 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一样的pythonexpression式:
1 if 0 is 0 else 3
如何在列表理解中使用它?
x if y else z
是您为每个元素返回的expression式的语法。 因此你需要:
[ x if x%2 else x*100 for x in range(1, 10) ]
混淆起因于你在第一个例子中使用filter ,而不是在第二个例子中。 在第二个例子中,只使用三元运算符expression式将每个值映射到另一个值。
使用filter,您需要:
[ EXP for x in seq if COND ]
没有filter,你需要:
[ EXP for x in seq ]
在你的第二个例子中,expression式是一个“复杂的”,恰好涉及到一个if-else
。
[x if x % 2 else x * 100 for x in range(1, 10) ]
只是另一个解决办法,希望有人会喜欢它:
使用:[False,True] [expression式]
>>> map(lambda x: [x*100, x][x % 2 != 0], range(1,10)) [1, 200, 3, 400, 5, 600, 7, 800, 9] >>>
你也可以用列表理解来做到这一点:
A=[[x*100, x][x % 2 != 0] for x in range(1,11)] print A
我能够做到这一点
>>> [x if x % 2 != 0 else x * 100 for x in range(1,10)] [1, 200, 3, 400, 5, 600, 7, 800, 9] >>>