📜  如何将一维数组转换为二维数组 - Python (1)

📅  最后修改于: 2023-12-03 15:38:44.969000             🧑  作者: Mango

如何将一维数组转换为二维数组 - Python

在Python中,可以通过numpy库中的reshape()函数来将一维数组转换为二维数组。

代码示例:
import numpy as np

# 生成一维数组
arr_1d = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])

# 将一维数组转换为二维数组
arr_2d = arr_1d.reshape(2, 5)

print("一维数组:")
print(arr_1d)

print("二维数组:")
print(arr_2d)

输出结果为:

一维数组:
[ 1  2  3  4  5  6  7  8  9 10]

二维数组:
[[ 1  2  3  4  5]
 [ 6  7  8  9 10]]

在示例代码中,首先通过numpy库中的array()函数生成一个一维数组arr_1d,然后调用reshape()函数将一维数组转换为二维数组,并将结果存储在arr_2d中。

reshape()函数的第一个参数是要转换成的行数,第二个参数是要转换成的列数。在本例中,一维数组被转换成了一个2行5列的二维数组。如果转换后的数组元素总数与原数组元素总数不匹配,则会抛出ValueError异常。

总结

通过numpy库中的reshape()函数,我们可以方便地将一维数组转换为二维数组。在使用时需要注意转换后的数组元素总数需要与原数组元素总数相等。