选择python中列表列表中的列表,类似于Mathematicas'All'

时间:2013-01-08 15:29:14

标签: python

我有一个格式列表:

cool_list = [[1, 2, 3, [4, 5, 6, 7]], [1, 2, 3, [4, 5, 6, 7]]]
在现实世界中,数字是不同的,并且有一些含义。

如何从[4, 5, 6, 7]中的所有数据中选择cool_list列表? 我认为mathematica等价物就像cool_list[[All],4]

修改
输出应该是

[[4, 5, 6, 7],
[4, 5, 6, 7]]

2 个答案:

答案 0 :(得分:2)

将[4,5,6,7]的所有列表放入列表中:

[x[3] for x in cool_list]

然后输出是:

[[4, 5, 6, 7], [4, 5, 6, 7]]

答案 1 :(得分:1)

如果我找对你,你会找

third_items = [x[3] for x in cool_list]
last_items = [x[-1] for x in cool_list]