flatten()函数用法

from numpy import *
a=mat([[1,2,3],[4,5,6]])
print(a)
print(a.flatten())

运行结果为:[[1 2 3]
[4 5 6]]
[[1 2 3 4 5 6]]

b=array([[1,2],[3,4],[5,6]])
print(b)
print(b.flatten())

运行结果为:[[1 2]
[3 4]
[5 6]]
1 2 3 4 5 6]

latten是numpy.ndarray.flatten的一个函数,即返回一个一维数组。

flatten只能适用于numpy对象,即array或者mat,普通的list列表不适用!。

a.flatten():a是个数组,a.flatten()就是把a降到一维,默认是按行的方向降 。
a.flatten().A:a是个矩阵,降维后还是个矩阵,矩阵.A(等效于矩阵.getA())变成了数组。

你可能感兴趣的:(flatten()函数用法)