小弟在處理一個很簡單的問題時,需要將 [1,2,[3,4]] 這樣的資料串列flatten為 [1,2,3,4]。
結果發現numpy的flatten方法並沒有辦法給出想要的結果:
list(np.array([1,2,[3,4]]).flatten())
我也嘗試的其他諸如 sum(list物件, []) 和 functools.reduce(operator.iconcat, list物件, [])
結果更糟,直接丟錯誤訊息說無法處理單一數字。
TypeError: 'int' object is not iterable
之所以會認為flatten就應該丟回 [1,2,3,4] 的結果是因為Mathematica就能正確處理:
Flatten[{1, 2, {3, 4}}]
Out[1]= {1, 2, 3, 4}
胃口被養壞了 XD
不知道還有什麼比較方便的方法可以讓numpy的flatten擲回想要的結果嗎? 謝謝!