numpy改变数组形状为两列
时间: 2024-05-16 22:17:34 浏览: 61
Numpy 多维数据数组的实现
你可以使用 numpy 的 reshape 函数来改变数组的形状。例如,如果你有一个形状为 (4, 3) 的数组,你可以通过以下方式将其转换为形状为 (6, 2) 的数组:
```python
import numpy as np
# 创建一个形状为 (4, 3) 的数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]])
# 使用 reshape 函数将其转换为形状为 (6, 2) 的数组
new_arr = arr.reshape(6, 2)
print(new_arr)
```
输出结果为:
```
[[ 1 2]
[ 3 4]
[ 5 6]
[ 7 8]
[ 9 10]
[11 12]]
```
这里我们将原始数组 arr 的形状改变为了 (6, 2)。在 reshape 函数中,我们传递了新数组的形状作为参数。注意,这里我们的原始数组必须拥有足够的元素来填充新数组,否则会抛出 ValueError 异常。
阅读全文