Python,使用列表推导

前端之家收集整理的这篇文章主要介绍了Python,使用列表推导前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

我有这样的代码

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

我想得到两个列表,第一个包含’a’的元素,其中a [] [1] = 1,第二个 – 元素,其中a [] [1] = 0所以

first_list = [[1,1]] 

second_list = [[3,0]]. 

我可以用两个列表理解来做这样的事情:

first_list = [i for i in a if i[1] == 1]

second_list = [i for i in a if i[1] == 0]

但也许存在其他(更多pythonic或更短)的方式来做到这一点?谢谢你的回答.

最佳答案
列表理解是非常pythonic和推荐的方式.你的代码很好.

猜你在找的Python相关文章