Numpy中sum函数的使用方法(Python自带sum函数)

Numpy中sum函数(Python自带sum函数)的作用是对元素求和。

  1. 无参时,所有全加;
  2. axis=0,按列相加;
  3. 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))

运行结果:

猜你喜欢

转载自blog.csdn.net/Sophia_11/article/details/84975009
今日推荐