Numpy:sum函数

Numpy的求和函数:sum

print("-----------一维数组使用sum函数是对全部元素求和------------------")
vector = numpy.array([5, 10, 15, 20])
print(vector.sum())  # 返回:50

print("-----------二维数组求和,一般与axis结合使用-----------------------")
matrix = numpy.array([
                [5, 10, 15], 
                [20, 25, 30],
                [35, 40, 45]
             ])
print(matrix.sum(axis=1))   # 对每行求和,返回:[ 30  75 120]

print(matrix.sum(axis=0))   # 对每列求和,返回:[60 75 90]

猜你喜欢

转载自www.cnblogs.com/wodexk/p/10309187.html