返回 NumPy 中掩码数组的对角线之和


要返回掩码数组元素的对角线之和,请在 NumPy 中使用 **ma.MaskedArray.trace()**。offset 参数是从主对角线偏移的对角线的偏移量。可以是正数也可以是负数。默认为 0。

axis 1 和 axis 2 是用作 2-D 子数组的第一轴和第二轴的轴,从中应获取对角线。默认为 a 的前两个轴。dtype 确定返回数组和累加器的数据类型,其中元素被求和。如果 dtype 的值为 None 且 a 为整数类型且精度小于默认整数精度,则使用默认整数精度。否则,精度与数组的精度相同。

步骤

首先,导入所需的库 -

import numpy as np
import numpy.ma as ma

使用 numpy.array() 方法创建包含 int 元素的数组 -

arr = np.array([[55, 85, 68, 84], [67, 33, 39, 53], [29, 88, 51, 37], [56, 45, 99, 85]])
print("Array...
", arr) print("
Array type...
", arr.dtype)

获取数组的维度 -

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

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

maskArr = ma.masked_array(arr, mask =[[1, 1, 0, 0], [ 0, 0, 1, 0], [0, 0, 0, 1], [0, 1, 0, 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)

要返回掩码数组元素的对角线之和,请在 NumPy 中使用 ma.MaskedArray.trace() -

res = maskArr.trace()
print("
Result..
.", res)

示例

import numpy as np
import numpy.ma as ma

# Create an array with int elements using the numpy.array() method
arr = np.array([[55, 85, 68, 84], [67, 33, 39, 53], [29, 88, 51, 37], [56, 45, 99, 85]])
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, 0], [ 0, 0, 1, 0], [0, 0, 0, 1], [0, 1, 0, 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 return the sum along diagonals of the masked array elements, use the ma.MaskedArray.trace() in Numpy res = maskArr.trace() print("
Result..
.", res)

输出

Array...
[[55 85 68 84]
[67 33 39 53]
[29 88 51 37]
[56 45 99 85]]

Array type...
int64

Array Dimensions...
2

Our Masked Array
[[-- -- 68 84]
[67 33 -- 53]
[29 88 51 --]
[56 -- 99 85]]

Our Masked Array type...
int64

Our Masked Array Dimensions...
2

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

Elements in the Masked Array...
16

Result..
. 169.0

更新于: 2022-02-22

101 次查看

启动你的 职业生涯

通过完成课程获得认证

开始
广告

© . All rights reserved.