numpy下的flatten()函式用法
即返回一個摺疊成一維的陣列。但是該函式只能適用於numpy物件,即array或者mat,普通的list列表是不行的。
例子:
1、用於array物件
1 2 3 4 5 6 7 8 |
from numpy import *
>>>a = array([[ 1 , 2 ],[ 3 , 4 ],[ 5 , 6 ]]) ###此時a是一個array物件
>>>a
array([[ 1 , 2 ],[ 3 , 4 ],[ 5 , 6 ]])
>>>a.flatten()
array([ 1 , 2 , 3 , 4 , 5 , 6 ])
|
2、用於mat物件
1 2 3 4 |
>>>
a = mat([[ 1 , 2 , 3 ],[ 4 , 5 , 6 ]])
>>>
a
matrix([[ 1 , 2 , 3 ],
[ 4 , 5 , 6 ]])<br>>>>
a.flatten()<br>matrix([[ 1 , 2 , 3 , 4 , 5 , 6 ]])<br>
|
3、但是該方法不能用於list物件
1 2 3 4 5 6 7 |
>>>
a = [[ 1 , 2 , 3 ],[ 4 , 5 , 6 ],[ 'a' , 'b' ]]
>>>
a
[[ 1 , 2 , 3 ],
[ 4 , 5 , 6 ],
[ 'a' , 'b' ]]
>>>
a.flatten() ###報錯
Traceback
(most recent call last):
File "<stdin>" ,
line 1 , in <module>
AttributeError: 'list'
|