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

numpy.MaskedArray.cumsum()返回給定軸上的蒙版數組元素的累積和。在計算過程中，蒙版值在內部設置為0。但是，將保存它們的位置，並且結果將在相同位置被屏蔽。

用法： numpy.ma.cumsum(axis=None, dtype=None, out=None)

參數：
axis:[int，可選]計算累積總和的軸。默認值(無)是計算展平數組上的總和。
dtype :[dtype，可選]返回數組的類型，以及與元素相乘的累加器的類型。
out :[ndarray，可選]將結果存儲到的位置。
->如果提供，則必須具有廣播輸入的形狀。
->如果未提供或沒有，則返回新分配的數組。

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

代碼1：

# Python program explaining 
# numpy.MaskedArray.cumsum() 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.cumsum     
# methods to masked array 
out_arr = mask_arr.cumsum()  
print ("cumulative sum of masked array along default axis:", out_arr)

輸出：

Input array: [[ 1  2]
 [ 3 -1]
 [ 5 -3]]
Masked array: [[-- 2]
 [-- -1]
 [5 -3]]
cumulative sum of masked array along default axis: [-- 2 -- 1 6 3]

代碼2：

# Python program explaining 
# numpy.MaskedArray.cumsum() 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.cumsum methods  
# to masked array 
out_arr1 = mask_arr.cumsum(axis = 0)  
print ("cumulative sum of masked array along 0 axis:", out_arr1) 
  
out_arr2 = mask_arr.cumsum(axis = 1)  
print ("cumulative sum of masked array along 1 axis:", out_arr2)

輸出：

Input array: [[1 0 3]
 [4 1 6]]
Masked array: [[1 0 3]
 [4 1 --]]
cumulative sum of masked array along 0 axis: [[1 0 3]
 [5 1 --]]
cumulative sum of masked array along 1 axis: [[1 1 4]

相關用法

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