python怎么取出二维数组的某一列

python

284

2023/9/22 19:26:19

栏目: 编程语言

可以使用列表推导式或者numpy库来取出二维数组的某一列。

假设有一个二维数组arr，要取出第i列，可以使用以下代码：

column = [row[i] for row in arr]

其中，i为要取出的列的索引。

如果已经安装了numpy库，可以使用numpy的切片操作来取出某一列。假设有一个二维数组arr，要取出第i列，可以使用以下代码：

import numpy as np
column = arr[:, i]

其中，i为要取出的列的索引。注意，numpy的切片操作中，":"表示全部元素。

使用numpy库的方法更加简洁和高效，尤其对于大型数组来说。

最新知识库