numpy数组取一列并且维度变成2
时间: 2023-05-20 08:01:39 浏览: 95
numpy按列连接两个维数不同的数组方式
可以使用numpy的切片操作来取一列,并使用reshape函数将维度变成2,代码如下:
```python
import numpy as np
# 创建一个3x3的二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 取第二列并将维度变成2
col = arr[:, 1].reshape(-1, 1)
print(col)
```
输出结果为:
```
array([[2],
[5],
[8]])
```
阅读全文