我有一個多維numpy 陣列，其形狀為 (5, 6192, 1)，因此基本上將 5 個長度為 6192 的陣列放入一個陣列中。
如何通過以下方式將所有陣列的元素添加到一個長度為 6192 的陣列中。例如，如果 5 個陣列看起來像
ar1 = [1,2,3...]
ar2 = [1,2,3...]
ar3 = [1,2,3...]
ar4 = [1,2,3...]
ar5 = [1,2,3...]
我希望我的最終陣列看起來像：
ar = [5,10,15,...]
因此，對于每個內部陣列，將每個相同位置的值添加到最終陣列的新值中，該值是該位置所有值的總和。
形狀應該是，我猜形狀（1,6192,1）。
uj5u.com熱心網友回復：
IIUC，只需使用numpy.sum：
ar1 = [1,2,3]
ar2 = [1,2,3]
ar3 = [1,2,3]
ar4 = [1,2,3]
ar5 = [1,2,3]
arrays = [ar1, ar2, ar3, ar4, ar5]
ar = np.sum(arrays, axis=0)
輸出：array([ 5, 10, 15])
如果你描述的形狀真的是正確的：
arr = np.array(arrays).reshape((5, 3, 1))
print(arr.shape)
# (5, 3, 1)
ar = np.sum(arr, axis=0)[None,:]
print(ar.shape)
# (1, 3, 1)
轉載請註明出處，本文鏈接：https://www.uj5u.com/qiye/435732.html
上一篇：我的排序中如何有第三個條件？