将2D NumPy数组拆分为两个1D数组的最简单方法是使用NumPy的flatten()
方法。该方法将多维数组转换为一维数组。
以下是示例代码:
import numpy as np
# 创建一个2D数组
arr_2d = np.array([[1, 2, 3], [4, 5, 6]])
# 使用flatten()方法将2D数组转换为1D数组
arr_1d_1 = arr_2d.flatten()
# 或者可以使用reshape()方法指定目标形状
arr_1d_2 = arr_2d.reshape(-1)
# 打印结果
print("1D数组1:", arr_1d_1)
print("1D数组2:", arr_1d_2)
输出结果:
1D数组1: [1 2 3 4 5 6]
1D数组2: [1 2 3 4 5 6]
这样,你就可以将2D NumPy数组拆分为两个1D数组。
领取专属 10元无门槛券
手把手带您无忧上云