NumPy如何改变数组的形状

numpy

1058

2024/5/16 12:49:12

栏目: 编程语言

NumPy可以通过reshape()方法来改变数组的形状。这个方法会返回一个新的具有指定形状的数组，但是原始数组的数据不会被复制，而是通过改变数组的步长来实现形状的改变。

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
new_arr = arr.reshape((3, 2))

print(new_arr)

这将输出:

[[1 2]
 [3 4]
 [5 6]]

在reshape()方法中，传入一个元组作为参数，元组的每个元素表示新数组的每个维度的大小。需要注意的是，新数组的元素个数必须和原数组的元素个数一致。

最新知识库