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

61次阅读
没有评论

共计 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 网 – 提供最优质的资源集合!

正文完
 
丸趣
版权声明:本站原创文章,由 丸趣 2023-12-22发表,共计366字。
转载说明:除特殊说明外本站除技术相关以外文章皆由网络搜集发布,转载请注明出处。
评论(没有评论)