python二维数组取某一列
时间: 2023-06-06 22:05:02 浏览: 171
python中实现数组和列表读取一列的方法
可以使用列表推导式或numpy库的方法进行操作。假设二维数组为arr,要取第i列,可以使用下面两种方法:
1. 列表推导式
```
col = [arr[j][i] for j in range(len(arr))]
```
2. numpy库
```
import numpy as np
col = np.array(arr)[:, i].tolist()
```
这两种方法都可以让你得到一个列表,其中存储了二维数组中第i列的所有元素。
阅读全文