numpy数组增加维度
时间: 2023-10-01 19:08:27 浏览: 106
Python numpy 11修改数组的维度.mp4
要将numpy数组增加一个维度,你可以使用numpy的`expand_dims`函数。它可以在指定的位置插入一个新的维度。
以下是使用`expand_dims`函数将一维数组转换为二维数组的示例:
```python
import numpy as np
# 创建一个一维数组
arr = np.array([1, 2, 3, 4, 5])
# 使用expand_dims函数在第二个维度插入一个新维度
new_arr = np.expand_dims(arr, axis=1)
print(new_arr.shape) # 输出:(5, 1)
```
在这个例子中,我们在原始数组的第二个维度插入了一个新的维度,从而将一维数组转换为了一个列向量的二维数组。`axis=1`参数表示插入的位置是第二个维度。
你也可以在其他位置插入新的维度,只需要根据需要调整`axis`参数的值即可。
阅读全文