python怎么取出二维数组的某一列
lewis
2018-06-11
24次阅读
可以使用列表推导式或者numpy库来取出二维数组的某一列。
- 使用列表推导式:
假设有一个二维数组arr,要取出第i列,可以使用以下代码:
column = [row[i] for row in arr]
其中,i为要取出的列的索引。
- 使用numpy库:
如果已经安装了numpy库,可以使用numpy的切片操作来取出某一列。假设有一个二维数组arr,要取出第i列,可以使用以下代码:
import numpy as np
column = arr[:, i]
其中,i为要取出的列的索引。注意,numpy的切片操作中,":"表示全部元素。
使用numpy库的方法更加简洁和高效,尤其对于大型数组来说。

发表评论:
◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。