我有一個分別具有 (3,n,m) 形狀的 R、G、B 3D 陣列，我想創建一個每個元素代表 (r,g,b) 的元組串列。我嘗試重塑和轉置，但它沒有按我預期的那樣作業
import numpy as np
arr = np.array(
[[[0, 1],
[2, 3]],
[[4, 5],
[6, 7]],
[[8, 9],
[10,11]]]
)
我想創建這樣的串列：
[(0, 4, 8), (1, 5, 9), (2, 6, 10), (3, 7, 11)]
uj5u.com熱心網友回復：
嘗試vstack并T使用tolist：
>>> np.vstack(arr.T).tolist()
[[0, 4, 8], [2, 6, 10], [1, 5, 9], [3, 7, 11]]
>>>
如果你想要元組：
>>> list(map(tuple, np.vstack(arr.T)))
[(0, 4, 8), (2, 6, 10), (1, 5, 9), (3, 7, 11)]
>>>
轉載請註明出處，本文鏈接：https://www.uj5u.com/ruanti/317013.html
上一篇：單獨計算求和引數