numpy二维数组添加一维
时间: 2023-08-29 20:06:04 浏览: 137
Numpy将二维数组添加到空数组的实现
可以使用`numpy.newaxis`来将一个二维数组添加一维。例如,将形状为`(3,4)`的二维数组添加一维可以这样做:
```python
import numpy as np
arr = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])
# 添加一维
new_arr = arr[:, np.newaxis, :]
print(new_arr.shape) # (3, 1, 4)
```
上述代码中,`[:, np.newaxis, :]`表示在第二个维度插入一个新维度。这将把原来的二维数组变成一个三维数组,新的维度的长度为1。
阅读全文