在 NumPy 中返回掩码数组的掩码

要返回掩码数组的掩码，请在 Python NumPy 中使用 **ma.getmask()** 方法。如果 a 是一个 MaskedArray 且掩码不是 nomask，则返回 a 的掩码作为 ndarray，否则返回 nomask。要保证与 a 形状相同的布尔值的完整数组，请使用 getmaskarray。

掩码数组是标准 numpy.ndarray 和掩码的组合。掩码要么是 nomask，表示关联数组的任何值均无效，要么是布尔值数组，用于确定关联数组的每个元素的值是否有效。

步骤

首先，导入所需的库 -

import numpy as np
import numpy.ma as ma

使用 numpy.arange() 方法创建一个包含 int 元素的 4x4 数组 -

arr = np.arange(16).reshape((4,4))
print("Array...
", arr)
print("
Array type...
", arr.dtype)

获取数组的维度 -

print("
Array Dimensions...
",arr.ndim)
print("
Our Array type...
", arr.dtype)

获取数组的形状 -

print("
Our Masked Array Shape...
",arr.shape)

获取数组的元素数量 -

print("
Elements in the Masked Array...
",arr.size)

创建一个掩码数组 -

arr = ma.array(arr)
arr[0, 1] = ma.masked
arr[1, 1] = ma.masked
arr[2, 1] = ma.masked
arr[2, 2] = ma.masked
arr[3, 0] = ma.masked
arr[3, 2] = ma.masked
arr[3, 3] = ma.masked

沿特定轴计算掩码元素的数量 -

print("
The number of masked elements...
",ma.count_masked(arr, axis = 1))

要返回掩码数组的掩码，请在 Python NumPy 中使用 ma.getmask() 方法 -

print("
Result (mask of a masked array)...
",ma.getmask(arr))

示例

import numpy as np
import numpy.ma as ma

# Creating a 4x4 array with int elements using the numpy.arange() method
arr = np.arange(16).reshape((4,4))
print("Array...
", arr)
print("
Array type...
", arr.dtype)

# Get the dimensions of the Array
print("
Array Dimensions...
",arr.ndim)
print("
Our Array type...
", arr.dtype)

# Get the shape of the Array
print("
Our Masked Array Shape...
",arr.shape)

# Get the number of elements of the Array
print("
Elements in the Masked Array...
",arr.size)

# Create a masked array
arr = ma.array(arr)
arr[0, 1] = ma.masked
arr[1, 1] = ma.masked
arr[2, 1] = ma.masked
arr[2, 2] = ma.masked
arr[3, 0] = ma.masked
arr[3, 2] = ma.masked
arr[3, 3] = ma.masked

# Count the number of masked elements along specific axis
print("
The number of masked elements...
",ma.count_masked(arr, axis = 1))

# To return the mask of a masked array, use the ma.getmask() method in Python Numpy
print("
Result (mask of a masked array)...
",ma.getmask(arr))

输出

Array...
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]
[12 13 14 15]]

Array type...
int64

Array Dimensions...
2

Our Array type...
int64

Our Masked Array Shape...
(4, 4)

Elements in the Masked Array...
16

The number of masked elements...
[1 1 2 3]

Result (mask of a masked array)...
[[False True False False]
[False True False False]
[False True True False]
[ True False True True]]

AmitDiwan

更新于: 2022年2月21日

1K+ 次查看

开启您的 职业生涯

通过完成课程获得认证

开始学习