Python numpy.ma.compress_cols()函数

这个numpy内置的函数可以抑制二维数组中包含屏蔽值的整列。

语法: numpy.ma.compress_cols(arr)

参数 :

arr : [array_like, MaskedArray]

1.该参数持有要操作的数组。
2.该数组必须是一个二维数组。
3.如果没有数组元素被屏蔽，arr被解释为一个MaskedArray，屏蔽设置为nomask。

返回 : 返回压缩后的数组。

下面是上述函数的实现。

示例 1:

# importing numpy as geek
import numpy as geek
  
# defining an array with mask
arr = geek.ma.array(geek.arange(6).reshape(2, 3),
                    mask=[[1, 0, 0], [0, 0, 0]])
  
# applying mask to array elements
gfg = geek.ma.compress_cols(arr)
  
print(gfg)

输出 :

[[1 2]
 [4 5]]

示例 2:

# importing numpy as geek
import numpy as geek
  
# defining array
arr = geek.ma.array(geek.arange(9).reshape(3, 3), mask=[
                    [1, 0, 0], [1, 0, 0], [0, 0, 0]])
  
# applying mask to array elements
gfg = geek.ma.compress_cols(arr)
  
print(gfg)

输出 :

[[1 2]
 [4 5]
 [7 8]]