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

python

513

2023/9/8 4: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]]

最新知识库