numpy_flatten函数

numpy的flatten函数用于将数组降至1维。

import numpy as np

a = np.array([[1, 2], [3, 4]])
print("a =", a)
b = a.flatten()
print("b =", b)

执行结果：

a = [[1 2]
     [3 4]]
b = [1 2 3 4]
版权声明：本文为fukangwei_lite原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。