我有以下 3Dnumpy.array形狀 (2,3,3)：
import numpy as np
arr = np.array(((((1,2,3),(9,8,7),(6,5,4))),
(((10,20,30),(90,80,70),(60,50,40)))))
我想對第一個維度求和并獲得一個形狀為 (3,3) 的二維陣列。
預期輸出：
array([[11, 22, 33],
[99, 88, 77],
[66, 55, 44]])
我知道我可以迭代第一維中的元素并求和，如下所示：
for el in range(len(arr)):
if el == 0:
arr_sum = arr[el]
else:
arr_sum = arr[el]
但是還有另一種選擇嗎？
提前致謝。
uj5u.com熱心網友回復：
您可以使用NumPy.sum方法。
import numpy as np
arr = np.array(((((1,2,3),(9,8,7),(6,5,4))),(((10,20,30),(90,80,70),(60,50,40)))))
print(np.sum(arr, axis=0))
轉載請註明出處，本文鏈接：https://www.uj5u.com/net/340817.html
上一篇：基于多維陣列生成標記
下一篇：docker容器之間的TCP連接