當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python Numpy MaskedArray.sum()用法及代碼示例


numpy.MaskedArray.median()函數用於計算給定軸上的蒙版數組元素的總和。

用法: numpy.ma.sum(arr, axis=None, dtype=None, out=None, keepdims=False)

參數:


arr :[ndarray]輸入掩碼數組。
axis :[int,可選]計算總和的軸。默認值(無)是在展平的數組上計算總和。
dtype : [dtype,可選]返回數組的類型,以及與元素相乘的累加器的類型。
out : [ndarray,可選]將結果存儲到的位置。
->如果提供,則必須具有廣播輸入的形狀。
->如果未提供或沒有,則返回新分配的數組。
keepdims :[bool,可選]如果將其設置為True,則縮小的軸將保留為尺寸為1的尺寸。使用此選項,結果將針對輸入數組正確廣播。

返回:[sum_along_axis,ndarray]除非指定out,否則將返回保存結果的新數組,在這種情況下,將返回對out的引用。

代碼1:

# Python program explaining 
# numpy.MaskedArray.sum() method  
    
# importing numpy as geek   
# and numpy.ma module as ma  
import numpy as geek  
import numpy.ma as ma  
    
# creating input array   
in_arr = geek.array([[1, 2], [ 3, -1], [ 5, -3]]) 
print ("Input array : ", in_arr)  
    
# Now we are creating a masked array.  
# by making  entry as invalid.   
mask_arr = ma.masked_array(in_arr, mask =[[1, 0], [ 1, 0], [ 0, 0]])  
print ("Masked array : ", mask_arr)  
    
# applying MaskedArray.sum     
# methods to masked array 
out_arr = ma.sum(mask_arr)  
print ("sum of masked array along default axis : ", out_arr) 
輸出:
Input array :  [[ 1  2]
 [ 3 -1]
 [ 5 -3]]
Masked array :  [[-- 2]
 [-- -1]
 [5 -3]]
sum of masked array along default axis :  3

代碼2:

# Python program explaining 
# numpy.MaskedArray.sum() method  
     
# importing numpy as geek   
# and numpy.ma module as ma  
import numpy as geek  
import numpy.ma as ma  
     
# creating input array  
in_arr = geek.array([[1, 0, 3], [ 4, 1, 6]])  
print ("Input array : ", in_arr) 
      
# Now we are creating a masked array.  
# by making one entry as invalid.   
mask_arr = ma.masked_array(in_arr, mask =[[ 0, 0, 0], [ 0, 0, 1]])  
print ("Masked array : ", mask_arr)  
     
# applying MaskedArray.sum methods  
# to masked array 
out_arr1 = ma.sum(mask_arr, axis = 0)  
print ("sum of masked array along 0 axis : ", out_arr1) 
  
out_arr2 = ma.sum(mask_arr, axis = 1)  
print ("sum of masked array along 1 axis : ", out_arr2)
輸出:
Input array :  [[1 0 3]
 [4 1 6]]
Masked array :  [[1 0 3]
 [4 1 --]]
sum of masked array along 0 axis :  [5 1 3]
sum of masked array along 1 axis :  [4 5]


相關用法


注:本文由純淨天空篩選整理自jana_sayantan大神的英文原創作品 Numpy MaskedArray.sum() function | Python。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。