np.array增加一个维度
时间: 2023-07-04 13:23:32 浏览: 116
numpy给array增加维度np.newaxis的实例
可以使用NumPy中的`np.newaxis`来增加一个维度。具体来说,如果你有一个形状为`(n,)`的一维数组,可以使用以下代码将其转换为形状为`(n, 1)`的二维数组:
```
import numpy as np
a = np.array([1, 2, 3, 4])
a = a[:, np.newaxis]
print(a)
```
输出结果为:
```
array([[1],
[2],
[3],
[4]])
```
你也可以使用`np.expand_dims`函数达到相同的效果,例如:
```
a = np.array([1, 2, 3, 4])
a = np.expand_dims(a, axis=1)
print(a)
```
输出结果同上。
阅读全文