【Python】numpy陣列轉為一維陣列：array.flatten()範例

一、flatten函數簡介

python的flatten()可以將array轉換為一維陣列，預設是order=’C’，以列為主，將每一列資料攤出來變成一維。

array.flatten([order])
[order] : 選填。order='C':以列為主；order='F':以欄為主

二、flatten函數範例

(一) order=’C’轉換

import numpy as np
a = np.array([ [1,2,3] ,[4,5,6]])
a.flatten('C')

結果 :
array([1, 2, 3, 4, 5, 6])

(二) order=’F’轉換

import numpy as np
a = np.array([ [1,2,3] ,[4,5,6]])
a.flatten('F')

結果 :
array([1, 4, 2, 5, 3, 6])