NumPy 一维数组转换为二维数组:reshape 函数示例
要将一维数组 a 转换为两行的二维数组,可以使用 NumPy 库中的 reshape 函数。以下是一个示例代码:
import numpy as np
a = np.array([1, 2, 3, 4])
b = a.reshape(2, 2)
print(b)
在上述代码中,我们首先创建了一个一维数组 'a',其中包含了元素 '[1, 2, 3, 4]'。然后,我们使用 'reshape' 函数将数组 'a' 转换为两行的二维数组 'b'。最后,我们打印输出数组 'b'。
运行上述代码,将会得到如下输出:
[[1 2]
[3 4]]
这里的输出显示了一个两行两列的数组,即 '[[1, 2], [3, 4]]'。通过使用 'reshape' 函数,我们成功地将一维数组 'a' 转换为了两行的二维数组。
原文地址: https://www.cveoy.top/t/topic/YNg 著作权归作者所有。请勿转载和采集!