[問題] Numpy的flatten方法結果似乎有錯怎麼解?

[dosiris (希望大家開心)]

小弟在處理一個很簡單的問題時，需要將 [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擲回想要的結果嗎？ 謝謝！

--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 111.250.78.225 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/Python/M.1682378493.A.BAC.html
※ 編輯: dosiris (111.250.78.225 臺灣), 04/25/2023 07:22:56
→ klamathu: 好奇這個[1, 2, [3, 4]]的Dimension是
→ dosiris: numpy回應shape是 (3,)
推 klamathu: itertools.chain()有符合大大的需求嗎？
一樣出現 TypeError: 'int' object is not iterable

→ lycantrope: ...因為numpy直接把它存成object
→ lycantrope: 應該有跳warning吧..
list(np.array([1,2,[3,4]]).flatten()) 沒跳出warning 但丟回的結果就是原樣

Out[31]: [1, 2, [3, 4]]
※ 編輯: dosiris (111.250.78.225 臺灣), 04/25/2023 09:26:49
→ lycantrope: https://pastebin.com/pPd5Fsus
→ lycantrope: 你可以看一下np.array([1,2,[3,4]])的dtype
→ dosiris: 謝謝分享！ 看來只能自訂指令解決了。學到一課
推 a78998042a: import more_itertools; nested_list=[1, 2, [3, 4]]
→ a78998042a: result=list(more_itertools.collapse(nested_list))
→ dosiris: 謝謝樓上！原來還有這招，學到了！