我有一个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))
这并没有给我所需的输出。 有什么建议吗?
答案 0 :(得分:2)
你可以做到
clf_prob.sum(axis=1)
这将取特定轴的总和,在本例中为行。
答案 1 :(得分:0)
numpy.sum()应该在您的情况下完成工作
答案 2 :(得分:0)
另一种可能性是使用numpy的ufunc:
np.sum.reduce(clf_prob)
这将给出第一个轴上的总和。您还可以使用axis
参数在另一个轴上求和。