温馨提示×

python一维数组怎么转二维数组

小亿
159
2023-08-18 04:20:29
栏目: 编程语言

可以使用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]]

0