Python，使用列表理解

Question

我有这样的代码:

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

我想得到两个列表，第一个包含 'a' 的元素，其中 a[][1] = 1，第二个 - 元素 a[][1] = 0。所以

first_list = [[1, 1], [2, 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 或更短的)方法来做到这一点？感谢您的回答。

Answer 1

列表推导非常 pythonic 和这样做的推荐方式。您的代码没问题。