壓扁變長數組列表

Question

如何轉換這樣的：

[2, 4, array([ 3.]), array([ 4.]), array([ 5., 4.])] 

到

[2,4,3,4,5,4] 

我試圖尋找，但該解決方案

[array([ 2.]), array([ 4.]), array([ 3.]), array([ 4.]), array([ 5., 4.])] 

和

工作

[[2],[4],[3],[4],[5,4]] 

Answer 1

import itertools 

a = [2, 4, array([ 3.]), array([ 4.]), array([ 5., 4.])] 
list(itertools.chain.from_iterable(asarray(b).ravel() for b in a)) 

Answer 2

我喜歡用sum來展平東西。在你的情況下，你首先需要將數組轉換爲列表，並將數字轉換爲數字列表。因此，如果陣列在：

sum((list(x) if hasattr(x, '__iter__') else [x] for x in a), [])