在python中匯總numpy數組的元素

Question

我有一個numpy數組

>>> clf_prob.dtype()

array([[ 0.05811791,  0.06526527,  0.06024136, ...,  0.06972481],
       [ 0.06093686,  0.06357167,  0.06462331, ...,  0.06999094],
       [ 0.08188396,  0.08504034,  0.0820972 , ...,  0.08487802],
       [ 0.05197106,  0.0786195 ,  0.15669477, ...,  0.0893244]])

我試圖添加這些數組的元素，這樣我的輸出將是：

[[0.05811791 + 0.06526527 + 0.06024136 +...+ 0.06972481],
[0.06093686 + 0.06357167 + 0.06462331 +...+0.06999094],
[0.08188396 + 0.08504034 + 0.0820972 + ...+  0.08487802],
[0.05197106 + 0.0786195  + 0.15669477+ ...+ 0.0893244]]

我試着做

sum(map(sum, clf_prob))

這沒有給我想要的輸出。 有什么建議嗎？

Answer 1

你可以做

clf_prob.sum(axis=1)

這將取某個軸上的總和，在本例中為行。

Answer 2

預期軸（1）上的numpy.sum（）應該適合您的情況

Answer 3

另一種可能性是使用numpy的ufunc：
np.sum.reduce(clf_prob)
這將給出第一個軸上的總和。 您還可以使用axis參數對另一個軸求和。