python将一维数组变二维

可以使用NumPy库中的reshape函数将一维数组转换为二维数组。例如，将长度为6的一维数组转换为2行3列的二维数组：

import numpy as np

arr1d = np.array([1, 2, 3, 4, 5, 6])
arr2d = arr1d.reshape(2, 3)

print(arr2d)

输出结果为：

[[1 2 3]
 [4 5 6]]

相关问题

python将一维数组变成二维数组

将一维数组变成二维数组可以使用numpy库中的reshape()函数。假设原数组为a，我们可以使用以下代码将其变成一个5行2列的二维数组：

import numpy as np
a = np.array([1,2,3,4,5,6,7,8,9,10])
b = a.reshape(5,2)
print(b)

输出结果为：

[[ 1  2]
 [ 3  4]
 [ 5  6]
 [ 7  8]
 [ 9 10]]

其中，reshape()函数的参数表示新数组的形状，也就是新数组的行数和列数。在上面的例子中，a数组中有10个元素，我们将其变成了5行2列的二维数组。需要注意的是，新数组的元素个数必须和原数组的元素个数相同，否则会抛出ValueError异常。

python将二维数组变为一维数组

你可以使用numpy库中的flatten方法将二维数组变为一维数组。例如：

import numpy as np

# 创建一个二维数组
arr_2d = np.array([[1, 2], [3, 4]])

# 使用flatten方法将二维数组变为一维数组
arr_1d = arr_2d.flatten()

print(arr_1d) # 输出 [1 2 3 4]

你也可以使用reshape方法将二维数组变为一维数组。例如：

import numpy as np

# 创建一个二维数组
arr_2d = np.array([[1, 2], [3, 4]])

# 使用reshape方法将二维数组变为一维数组
arr_1d = arr_2d.reshape(-1)

print(arr_1d) # 输出 [1 2 3 4]

这里的`-1`表示自动计算数组的长度，即根据原始数组的维度来自动计算新的数组长度。