如何在python中获取列表的子集?

时间:2017-04-29 20:51:48

标签: python list subset

我想要的是:

>>> [i if i!= 0 for i in [0,1,2,3]]
[1,2,3]

就像

一样
>>> [i for i in [1,2,3,4]]
[1,2,3,4]

什么是没有产生语法错误的简单解决方案?

编辑:假设我不想使用for循环并将所有元素附加到新列表中。

4 个答案:

答案 0 :(得分:3)

使用[i for [0,1,2,3] if i!= 0]来获取

[1,2,3]

答案 1 :(得分:1)

您最后可以添加if

[i for i in [0,1,2,3] if i!= 0]

答案 2 :(得分:1)

[i for i in [0,1,2,3] if 1 != 0]

答案 3 :(得分:1)

只需将if i != 0放在列表理解的末尾,如下所示:

[i for i in [0,1,2,3] if i!=0]