如何访问python中列表中的列

时间:2017-06-05 00:24:37

标签: python arrays numpy multiple-columns

我在python中有一个2D数组,由列表列表建模,我想提取列。我做了一个快速的研究,我发现了一种使用numpy数组的方法。问题是我不想使用numpy 所以我不想将我的列表列表转换为numpy数组,然后使用[:,1]语法。我尝试在正常的列表列表中使用它,但它显示错误,所以它不可能。我要求列表列表类似的东西,而不必遍历每个元素(在numpy数组中,使用[:,1]语法访问列比通过迭代数组元素更快)。

我发现了这个link,但它又建议在没有快捷方式的情况下迭代元素。

提前致谢。

2 个答案:

答案 0 :(得分:18)

在处理列表列表时,列表推导是您的朋友:

In [111]: alist
Out[111]: 
[[0, 1, 2, 3, 4, 5],
 [6, 7, 8, 9, 10, 11],
 [12, 13, 14, 15, 16, 17],
 [18, 19, 20, 21, 22, 23]]
In [112]: [row[1] for row in alist]
Out[112]: [1, 7, 13, 19]

还有一个方便的成语'用于转置嵌套列表,转换'列'进入'行':

In [113]: tlist = list(zip(*alist))
In [114]: tlist
Out[114]: 
[(0, 6, 12, 18),
 (1, 7, 13, 19),
 (2, 8, 14, 20),
 (3, 9, 15, 21),
 (4, 10, 16, 22),
 (5, 11, 17, 23)]
In [115]: tlist[1]
Out[115]: (1, 7, 13, 19)

答案 1 :(得分:2)

您可以转置嵌套列表(即将您的'列'转换为'行'): https://stackoverflow.com/a/21444360/7306999