Numpy中sum函数(Python自带sum函数)的作用是对元素求和。
- 无参时,所有全加;
- axis=0,按列相加;
- axis=1,按行相加;
下边通过例子来说明其用法:
#!/usr/bin/env python
# -*- coding:utf-8 -*-
from numpy import *
import numpy as np
a = array([[1, 2, 3], [4, 5, 6]])
# Python自带sum函数相加
SUM = sum(a) # 所有元素求和
SUM_0 = sum(a, 0) # 按照列求和
SUM_1 = sum(a, 1) # 按照行求和
print("SUM:"+str(SUM))
print("SUM_0:"+str(SUM_0))
print("SUM_1:"+str(SUM_1))
# numpy中sum函数相加
np_SUM = np.sum(a) # 所有元素求和
print("np_SUM:"+str(np_SUM))
np_SUM_0 = np.sum(a, axis=0) # 按照列求和
print("np_SUM_0:"+str(np_SUM_0))
np_SUM_1 = np.sum(a, axis=1) # 按照行求和
print("np_SUM_1:"+str(np_SUM_1))
运行结果: