如何在Python中将输入数据重塑成二维数组

将输入数据重塑成二维数组可以使用NumPy库中的reshape函数。假设输入数据为一维数组data，可以按如下方式重塑成二维数组arr：

import numpy as np

# 假设 data 是一维数组
arr = np.reshape(data, (-1, n))

# 其中 n 为每行元素个数，-1 表示自动计算。例如，将长度为 6 的一维数组重塑为 2 行 3 列的二维数组：
data = np.array([1, 2, 3, 4, 5, 6])
arr = np.reshape(data, (2, 3))
print(arr)

# 输出：
# [[1 2 3]
#  [4 5 6]]

注意，重塑后的二维数组行数、列数与输入数据元素个数要匹配，否则会抛出ValueError异常。