Python列表列表->使元素n成为子列表

Question

如果我有清单清单，例如：

[[2, 0.84], [2, 0.75], [2, 0.63], [2, 0.6]]

有没有一种快速的方法来获取内部列表的所有第二个元素的列表？ 即，该示例将返回：

[0.84, 0.75, 0.63, 0.6]

我不确定这为什么行不通：

list[:][0]

但这只会在列表中返回一对值。

Answer 1

尝试：

>>> my_list = [[2, 0.84], [2, 0.75], [2, 0.63], [2, 0.6]]
>>> [item[1] for item in my_list]
[0.84, 0.75, 0.63, 0.6]

Answer 2

您的尝试list[:][0]不起作用，因为list[:]会复制list ， list[:][0]将获取副本的第一个子列表，而不是子列表的第一个元素。 另外，避免命名变量list因为它会遮盖内置list 。

这是一个为您提供所需输出的解决方案，但是正如@Slater Tyranus的回答所示，这不是最有效的方法。

In [34]: a = [[2, 0.84], [2, 0.75], [2, 0.63], [2, 0.6]]

In [36]: list(zip(*a)[1])
Out[36]: [0.84, 0.75, 0.63, 0.6]

Answer 3

Hasan在一般情况下是正确的，但是如果您通常在寻找这种切片，我建议您开始使用numpy，这实际上是为这种事情而做的。 特别：

>>> import numpy as np

>>> my_array = np.array([[2, 0.84], [2, 0.75], [2, 0.63], [2, 0.6]])
>>> my_array[:, 1]
[0.84, 0.75, 0.63, 0.6]

就我个人而言，我发现numpy切片更容易使用，并且它似乎与您对上述问题的直觉非常吻合。 而且，使用大型阵列时，速度可能会更快。 也就是说，对于这种大小的阵列，显然要慢得多，在下面发布时间结果：

>>> my_list = [[2, 0.84], [2, 0.75], [2, 0.63], [2, 0.6]]
>>> my_array = np.array(my_list)

>>> timeit.timeit("my_array[:, 1]", setup="from __main__ import my_array", number=10000)
0.016569137573242188

>>> timeit.timeit("[item[1] for item in my_list]", setup="from __main__ import my_list", number=10000)
0.006146907806396484

>>> timeit.timeit("list(zip(*my_list)[1])", setup="from __main__ import my_list", number=10000)
0.013128042221069336

为了显示更清晰的图片，我在一个较大的示例中运行了结果，以显示这些不同方法的缩放比例：

>>> my_array = np.random.rand(100,100)
>>> my_list = my_array.tolist()

>>> timeit.timeit("my_array[:, 1]", setup="from __main__ import my_array", number=10000)
0.019372940063476562

>>> timeit.timeit("[item[1] for item in my_list]", setup="from __main__ import my_list", number=10000)
0.07012009620666504

>>> timeit.timeit("list(zip(*my_list)[1])", setup="from __main__ import my_list", number=10000)
1.2308871746063232