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

寻技术 Python编程 前天 15:30 7

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

用微信“扫一扫”