使用NumPy计算掩码数组中未掩码元素的个数,沿给定轴


要计算沿给定轴的掩码数组中未掩码元素的个数,可以使用Python NumPy中的**ma.MaskedArray.count()**方法。“**axis**”参数用于设置轴。该方法返回一个与输入数组形状相同的数组,但指定的轴已移除。如果数组是0维数组,或者axis为None,则返回标量。

axis参数是执行计数的轴或轴。默认值None在输入数组的所有维度上执行计数。axis可以为负数,在这种情况下,它从最后一个轴到第一个轴进行计数。

如果keepdims参数设置为True,则减少的轴将保留在结果中,大小为一。使用此选项,结果将与数组正确广播。

步骤

首先,导入所需的库:

import numpy as np
import numpy.ma as ma

使用numpy.array()方法创建一个包含整数元素的数组:

arr = np.array([[65, 68, 81], [93, 33, 39], [73, 88, 51], [62, 45, 67]])
print("Array...
", arr) print("
Array type...
", arr.dtype)

获取数组的维度:

print("
Array Dimensions...
",arr.ndim)

创建一个掩码数组,并将其中一些标记为无效:

maskArr = ma.masked_array(arr, mask =[[1, 1, 0], [ 1, 0, 0], [0, 1, 0], [0, 1, 0]])
print("
Our Masked Array
", maskArr) print("
Our Masked Array type...
", maskArr.dtype)

获取掩码数组的维度:

print("
Our Masked Array Dimensions...
",maskArr.ndim)

获取掩码数组的形状:

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

获取掩码数组的元素个数:

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

要计算沿给定轴的掩码数组中未掩码元素的个数,可以使用ma.MaskedArray.count()方法。“axis”参数用于设置轴:

print("
Result...
",maskArr.count(axis = 1))

示例

# Python ma.MaskedArray - Count the non-masked elements of the masked array along the given axis

import numpy as np
import numpy.ma as ma

# Create an array with int elements using the numpy.array() method
arr = np.array([[65, 68, 81], [93, 33, 39], [73, 88, 51], [62, 45, 67]])
print("Array...
", arr) print("
Array type...
", arr.dtype) # Get the dimensions of the Array print("
Array Dimensions...
",arr.ndim) # Create a masked array and mask some of them as invalid maskArr = ma.masked_array(arr, mask =[[1, 1, 0], [ 1, 0, 0], [0, 1, 0], [0, 1, 0]]) print("
Our Masked Array
", maskArr) print("
Our Masked Array type...
", maskArr.dtype) # Get the dimensions of the Masked Array print("
Our Masked Array Dimensions...
",maskArr.ndim) # Get the shape of the Masked Array print("
Our Masked Array Shape...
",maskArr.shape) # Get the number of elements of the Masked Array print("
Elements in the Masked Array...
",maskArr.size) # To count the non-masked elements of the masked array along the given axis, use the ma.MaskedArray.count() method # The axis is set using the "axis" parameter print("
Result...
",maskArr.count(axis = 1))

输出

Array...
[[65 68 81]
[93 33 39]
[73 88 51]
[62 45 67]]

Array type...
int64

Array Dimensions...
2

Our Masked Array
[[-- -- 81]
[-- 33 39]
[73 -- 51]
[62 -- 67]]

Our Masked Array type...
int64

Our Masked Array Dimensions...
2

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

Elements in the Masked Array...
12
Result...
[1 2 2 2]

更新于:2022年2月3日

95 次浏览

启动你的职业生涯

完成课程获得认证

开始学习
广告