numpy中sum(axis=1)与sum(axis=0)的区别

import numpy
a = numpy.array([
    [[5,10,15],
    [10,20,30],
    [20,40,60]],
    
    [[5,10,15],
    [10,20,30],
    [20,40,60]]
])
#按行相加,此时第一、二行都是[5,10,15],[10,20,30],[20,40,60]
b = a.sum(axis=1)
#按列相加,此时第一列是[5,10,15],第二列是[10,20,30],第三列是[20,40,60]
c = a.sum(axis=0)

print(b)
print(c)

结果:

[[ 35  70 105]
 [ 35  70 105]]
[[ 10  20  30]
 [ 20  40  60]
 [ 40  80 120]]
总结:简单的理解,axis=1按行的方向相加,返回每个行的值;axis=0按列相加,返回每个列的值。


你可能感兴趣的:(机器学习)