flatten函數（python）（超級易懂~）

功能：將numpy數組展開為一維數組

一.     默認方向是行方向，加’a'也是行方向，但是加‘f'是列方向

import numpy as np
a = np.array([[1, 2],
[3, 4],
[5, 6],
[7, 8]])
m = a.flatten()
n = a.flatten('a')
k = a.flatten('f')
print(m)
print(n)
print(k)

輸出結果為：

二.   需要注意的是，flatten函數不能直接作用於列表 ！

比如，我們將列表改為如下：

a = [[1, 2],
[3, 4],
[5, 6],
[7, 8]]
m = a.flatten()
print(m)

這時候，系統就會報錯

 三.    那麼，對於列表應該怎麼做呢，我們可以改成numpy數組，也可以將代碼改為如下：

a = [[1, 2],
[3, 4],
[5, 6],
[7, 8]]
m = [x for i in a for x in i]
print(m)

代碼中的i就是列表中的各個列表了，x在這些小列表中逐漸取值組成了m

這時候，代碼的結果就變為：

如果還有哪兒不懂，歡迎評論區中提出