numpy 如何在python中访问列表的列表中的列

hrirmatl  于 2022-12-29  发布在  Python
关注(0)|答案(1)|浏览(110)

我在python中有一个2D数组,由一个列表列表建模,我想提取列。我做了一个快速的研究,我发现了一种使用numpy数组的方法。问题是我不想使用numpy所以我不想将我的列表列表转换成numpy数组,然后使用[:,1]语法。我尝试在一个普通的列表列表上使用它,但它显示了一个错误,所以这是不可能的。我要求对列表列表做类似的事情,而不必遍历每个元素(在numpy数组中,使用[:,1]语法访问列比迭代数组的元素更快)。
我找到了这个link,但它再次建议在没有快捷方式的情况下迭代元素。

gjmwrych

gjmwrych1#

列表解析在处理列表的列表时是你的朋友:

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)

相关问题