python数组不改变长度,增加维度
时间: 2024-01-25 15:18:55 浏览: 101
不改变长度删除数组中的元素
如果你希望增加维度,但不改变数组的长度,可以使用numpy库中的reshape方法或者numpy.newaxis。以一维数组为例,可以使用reshape方法将其转换为二维数组,或者使用numpy.newaxis将其转换为二维数组的行或列,而不改变数组的长度。
使用reshape方法:
```python
import numpy as np
a = np.array([1, 2, 3, 4])
b = a.reshape(2, 2)
print(b)
```
输出结果为:
```
[[1 2]
[3 4]]
```
使用numpy.newaxis:
```python
import numpy as np
a = np.array([1, 2, 3, 4])
b = a[:, np.newaxis]
print(b)
```
输出结果为:
```
[[1]
[2]
[3]
[4]]
```
使用numpy.newaxis将一维数组转换为二维数组的行或列,可以根据需要进行选择。例如,a[:, np.newaxis]将一维数组转换为二维数组的列,而a[np.newaxis, :]将一维数组转换为二维数组的行。
阅读全文