Python NumPy: 使用 reshape 将一维数组转换为二维数组
可以使用 NumPy 库中的 reshape 函数将一个一维数组转换为二维数组。reshape 函数接受一个元组作为参数,表示新数组的形状。如果新数组的形状无法满足原数组的大小,则会抛出异常。
示例代码:
import numpy as np
# 生成一个一维数组
arr1d = np.array([1, 2, 3, 4, 5, 6])
# 将一维数组转换为二维数组
arr2d = arr1d.reshape((2, 3))
# 打印结果
print(arr2d)
输出:
[[1 2 3]
[4 5 6]]
在上面的代码中,我们将一个一维数组 arr1d 转换为了一个 2 行 3 列的二维数组 arr2d。注意,reshape 函数并不会改变原数组的形状,而是返回一个新的数组。
原文地址: https://www.cveoy.top/t/topic/ozpB 著作权归作者所有。请勿转载和采集!