1. 程式人生 > >Numpy中sum函式的使用方法(Python自帶sum函式)

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))

執行結果: