python - 将 1D 数组 reshape 为矩阵 2D 矩阵

Question

我想将 1D 数组转换为 200 行 5 列的 2D 数组，但我的代码抛出一个错误:ValueError: 无法将大小为 1 的数组 reshape 为形状 (200,5)

import numpy as np
np.random.seed(25)
ar = np.random.randn(1000)
ar = ar * 100
ar = ar.astype('int8')
ar

到目前为止都很好，但接下来我想 reshape 这个大小为 1000 的数组，并将该数组 reshape 为 5 列、200 行的集合。返回这个新数组并用它替换原来的 ar 数组。所以我写了下面的代码，但是 reshape 不起作用(ValueError:无法将大小为1的数组 reshape 为形状(200,5))，我的代码有什么问题？

def reshape(my_array):
    ar = np.reshape(my_array,(200,5))
    ar

ar = reshape(ar)

Answer 1

我可以建议这个改变吗？它会工作，因为我认为你将一个错误的 ar 传递到函数中:

import numpy as np
np.random.seed(25)
ar = np.random.randn(1000)
ar = ar * 100
ar = ar.astype('int8')
ar

def reshape(my_array):
    ar = np.reshape(my_array,(200,5))
    return ar

new_ar = reshape(ar)