Python怎么将一个一维数组 data 的第一列提取出来,并将其转换为一个列向量
时间: 2024-02-06 10:10:48 浏览: 73
numpy:np.newaxis 实现将行向量转换成列向量
5星 · 资源好评率100%
你可以使用NumPy库来完成这个任务。首先,你需要将一维数组转换为二维数组,然后选择第一列并转置它。下面是代码示例:
```python
import numpy as np
# 创建一维数组
data = np.array([1, 2, 3, 4, 5])
# 将一维数组转换为二维数组
data = data[:, np.newaxis]
# 选择第一列并转置
col_1 = data[:, 0].T
print(col_1)
```
输出:
```
[[1]
[2]
[3]
[4]
[5]]
```
在这个代码示例中,`data[:, np.newaxis]`将一维数组转换为一个二维数组。然后,选择第一列并使用`.T`方法将其转置为一个列向量。
阅读全文