我在python中有一个2D数组,由一个列表列表建模,我想提取列。我做了一个快速的研究,我发现了一种使用numpy数组的方法。问题是我不想使用numpy所以我不想将我的列表列表转换成numpy数组,然后使用[:,1]语法。我尝试在一个普通的列表列表上使用它,但它显示了一个错误,所以这是不可能的。我要求对列表列表做类似的事情,而不必遍历每个元素(在numpy数组中,使用[:,1]语法访问列比迭代数组的元素更快)。我找到了这个link,但它再次建议在没有快捷方式的情况下迭代元素。
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)
1条答案
按热度按时间gjmwrych1#
列表解析在处理列表的列表时是你的朋友:
还有一个方便的“习惯用法”可以用来转置嵌套列表,将“列”转置为“行”: