当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。