numpy中flatten学习笔记

ndarray.flatten()

用法 用于返回一个折叠成一维的数组。该函数只能适用于numpy对象，即array或者mat，普通的list列表是不行的。

例子

# coding=utf-8
from numpy import *

a = array([[1, 2], [3, 4], [5, 6]])  #array对象
b=  mat([[1,2,3],[4,5,6]])            #mat对象
c = [[1, 2, 3], [4, 5, 6], ['a', 'b']]  #list对象
print(a.flatten())
print(b.flatten())
# print(c.flatten())  #错误
print([y for x in cfor y in x])  #list对象展开的正确做法

 

输出