python array reshape
时间: 2023-09-03 12:14:51 浏览: 201
Python库 | reshape-0.1.0.tar.gz
在Python中,你可以使用NumPy库中的reshape函数来改变数组的形状。reshape函数将数组重新排列为指定的形状,而不改变其数据。
下面是一个示例:
```python
import numpy as np
# 创建一个一维数组
arr = np.array([1, 2, 3, 4, 5, 6])
# 将数组reshape为2行3列的二维数组
reshaped_arr = arr.reshape(2, 3)
print(reshaped_arr)
```
输出:
```
[[1 2 3]
[4 5 6]]
```
在这个例子中,我们首先创建了一个包含6个元素的一维数组。然后,我们使用reshape函数将该数组重新排列为一个2行3列的二维数组。输出结果显示了二维数组的形状。
你还可以使用-1作为参数来自动计算某个维度的大小。例如,如果你想将一个包含12个元素的一维数组转换为3行4列的二维数组,可以这样做:
```python
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12])
reshaped_arr = arr.reshape(3, -1)
print(reshaped_arr)
```
输出:
```
[[ 1 2 3 4]
[ 5 6 7 8]
[ 9 10 11 12]]
```
在这个例子中,我们将第一个维度指定为3,而将第二个维度指定为-1。NumPy会自动计算出第二个维度的大小为4,以便保持总的元素数量不变。
希望这能帮到你!如果你还有其他问题,请随时提问。
阅读全文