Python Numpy MaskedArray.flatten()函数

numpy.MaskedArray.flatten()函数用于返回一个被折叠成一维的输入掩码数组的副本。

语法: numpy.ma.flatten(order='C')

参数:
order : [‘C’, ‘F’, ‘A’, ‘K’, optional] 是否以C（行主）、Fortran（列主）顺序进行平移，或者保留A的C/Fortran顺序。

返回 : [ ndarray] 一个输入数组的副本，被压扁为一维。

代码#1：

# Python program explaining
# numpy.MaskedArray.flatten() method 
    
# importing numpy as geek  
# and numpy.ma module as ma 
import numpy as geek 
import numpy.ma as ma 
    
# creating input array of 2 * 2  
in_arr = geek.array([[10, 20], [-10, 40]]) 
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 =[[ 1, 0], [ 0, 0]]) 
print ("Masked array : ", mask_arr) 
    
# applying MaskedArray.flatten methods to make  
# it a 1D flattened array
out_arr = mask_arr.flatten() 
print ("Output flattened masked array : ", out_arr)

输出:

Input array :  [[ 10  20]
 [-10  40]]
Masked array :  [[-- 20]
 [-10 40]]
Output flattened masked array :  [-- 20 -10 40]

代码#2：

# Python program explaining
# numpy.MaskedArray.flatten() 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([[[ 2e8, 3e-5]], [[ -4e-6, 2e5]]])
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 =[[[ 1, 0]], [[ 0, 0]]]) 
print ("Masked array : ", mask_arr) 
    
# applying MaskedArray.flatten methods to make  
# it a 1D masked array
out_arr = mask_arr.flatten(order ='F') 
print ("Output flattened masked array : ", out_arr)

输出:

Input array :  [[[ 2.e+08  3.e-05]]

 [[-4.e-06  2.e+05]]]
Masked array :  [[[-- 3e-05]]

 [[-4e-06 200000.0]]]
Output flattened masked array :  [-- -4e-06 3e-05 200000.0]