flatten()函数用法

flatten是numpy.ndarray.flatten的一个函数,即返回一个折叠成一维的数组。但是该函数只能适用于numpy对象,即array或者mat,普通的list列表是不行的。
a是个矩阵或者数组,a.flatten()就是把a降到一维,默认是按横的方向降

from numpy import *
a = array([[1,2],[3,4]])
print(a.flatten())
<<<[1 2 3 4]
>>> a.flatten('F')
array([1, 3, 2, 4])  # 按列排序
posted @ 2023-01-10 12:52  辛宣  阅读(167)  评论(0)    收藏  举报