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())变成了数组。

发布了40 篇原创文章 · 获赞 1 · 访问量 2058

猜你喜欢

转载自blog.csdn.net/qq_40660825/article/details/100185135