转换ndarray的维度
时间: 2024-04-08 20:35:06 浏览: 67
numpy库ndarray多维数组的维度变换方法(reshape、resize、swapaxes、flatten)
要转换一个NumPy数组(ndarray)的维度,可以使用`reshape()`函数。`reshape()`函数接受一个元组作为参数,用于指定新的度。
以下是一个示例代码,演示何转换一个NumPy数组的维度:
```python
import numpy as np
# 创建一个34的二维数组
arr = np.array([[, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12]])
# 转换为2x6的二维数组
new_arr = arr.reshape((2, 6))
# 打印转换后的数组
print(new_arr)
```
在这个示例中,我们创建了一个3x4的二维数组`arr`。然后,我们使用`reshape()`函数将其转换为一个2x6的二维数组`new_arr`。
请注意,转换后的数组的总元素数量必须与原数组相同。也就是说,新的形状中的维度乘积必须等于原数组中的维度乘积。
如果你想要将一个多维数组转换为一维数组,可以使用`flatten()`函数或`ravel()`函数。这两个函数都会将多维数组展平成一维数组,但有一些微小的差别。
希望这对你有所帮助!如果你还有其他问题,请随时提问。
阅读全文