共计 366 个字符,预计需要花费 1 分钟才能阅读完成。
可以使用 numpy 库中的 reshape 函数将一维数组转换为二维数组。假设一维数组名为 arr,需要转换为 m 行 n 列的二维数组,则可以使用如下代码:
import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
m = 3 # 行数
n = 3 # 列数
arr_2d = np.reshape(arr, (m, n))
print(arr_2d)
输出结果为:
[[1 2 3]
[4 5 6]
[7 8 9]]
也可以使用列表推导式来实现一维数组转二维数组的功能,如下所示:
arr = [1, 2, 3, 4, 5, 6, 7, 8, 9]
m = 3 # 行数
n = 3 # 列数
arr_2d = [arr[i:i+n] for i in range(0, len(arr), n)]
print(arr_2d)
输出结果为:
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
丸趣 TV 网 – 提供最优质的资源集合!
正文完