Numpy使用

1. NumPy 中的 .flatten()

NumPy 数组的 flatten() 方法返回原数组的副本，将多维数组展平为一维。

import numpy as np

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

# 展平数组
flattened = arr.flatten()

print("原数组：")
print(arr)
# 输出：
# [[1 2 3]
#  [4 5 6]]

print("\n展平后的数组：")
print(flattened)  # 输出：[1 2 3 4 5 6]
print(flattened.shape)  # 输出：(6,)