在NumPy中,可以使用flatten()
方法将二维数组转换为一维数组。
下面是一个示例:
import numpy as np
# 创建一个二维数组
arr_2d = np.array([[1, 2, 3], [4, 5, 6]])
# 将二维数组转为一维数组
arr_1d = arr_2d.flatten()
print(arr_1d)
输出结果:
[1 2 3 4 5 6]
另外,还可以使用reshape()
方法将二维数组转换为一维数组。
下面是使用reshape()
方法的示例:
import numpy as np
# 创建一个二维数组
arr_2d = np.array([[1, 2, 3], [4, 5, 6]])
# 使用reshape方法将二维数组转为一维数组
arr_1d = arr_2d.reshape(-1)
print(arr_1d)
输出结果与前面的示例相同:
[1 2 3 4 5 6]