从2D NumPy数组中找到一维数组？

import numpy as np

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

# 使用flatten()函数展开数组
arr_1d = arr_2d.flatten()

print(arr_1d)

[1 2 3 4 5 6]

import numpy as np

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

# 使用ravel()函数展开数组
arr_1d = arr_2d.ravel()

print(arr_1d)

[1 2 3 4 5 6]