返回两个掩码数组的点积,并设置是否在 Numpy 中传播掩码数据

numpyserver side programmingprogramming更新于 2025/9/3 6:22:17

要返回两个掩码数组的点积,请使用 Python Numpy 中的 ma​​.dot() 方法。"strict"参数设置是否传播掩码数据 (True) 或将其设置为 0 (False) 以进行计算。

此函数相当于考虑掩码值的 numpy.dot。strict 和 out 的位置与方法版本中的位置不同。为了保持与相应方法的兼容性,建议将可选参数仅视为关键字。在某些情况下,这可能是强制性的。

strict 参数设置是否传播掩码数据 (True) 或将其设置为 0 (False) 以进行计算。默认值为 False。传播掩码意味着,如果被掩码的值出现在一行或一列中,则整行或整列都被视为被掩码。

输出参数表明,它必须具有未使用时返回的精确类型。具体来说,它必须具有正确的类型,必须是 C 连续的,并且其 dtype 必须是 dot(a,b) 返回的 dtype。这是一个性能特性。因此,如果不满足这些条件,则会引发异常,而不是尝试保持灵活性。

步骤

首先,导入所需的库 −

import numpy as np
import numpy.ma as ma

使用 numpy.arange() 方法创建数组 1,一个包含 int 元素的 3x3 数组 −

arr1 = np.arange(9).reshape((3,3))
print("Array1...
", arr1) print("
数组类型...
", arr1.dtype)

# 创建掩码数组 1 −

arr1 = ma.array(arr1)

掩码数组 1 −

arr1[0, 1] = ma.masked
arr1[1, 1] = ma.masked

显示掩码数组 1 −

print("
掩码数组 1...
",arr1)

使用 numpy.arange() 方法 − 创建数组 2,另一个包含 int 元素的 3x3 数组

arr2 = np.arange(9).reshape((3,3))
print("
Array2...
", arr2) print("
数组类型...
", arr2.dtype)

创建掩码数组 2

arr2 = ma.array(arr2)

掩码数组 2 −

arr2[2, 1] = ma.masked
arr2[2, 2] = ma.masked

显示掩码数组 2 −

print("
掩码数组 2...
",arr2)

要返回两个掩码数组的点积,请使用 Python Numpy 中的 ma.dot() 方法。"严格"参数设置是否传播屏蔽数据 (True) 或将计算结果设置为 0 (False) −

print("
点积结果...
",np.ma.dot(arr1, arr2, strict=True))

示例

import numpy as np
import numpy.ma as ma

# 数组 1
# 使用 numpy.arange() 方法创建一个包含 int 元素的 3x3 数组
arr1 = np.arange(9).reshape((3,3))
print("Array1...
", arr1) print("
数组类型...
", arr1.dtype) # 获取数组的维度 print("
数组维度...
",arr1.ndim) # 获取数组的形状 print("
我们的数组形状...
",arr1.shape) # 获取数组元素的数量 print("
数组中的元素...
",arr1.size) # 创建一个掩码数组 arr1 = ma.array(arr1) # 掩码数组 1 arr1[0, 1] = ma.masked arr1[1, 1] = ma.masked # 显示掩码数组 1 print("
掩码数组 1...
",arr1) # 数组 2 # 使用 numpy.arange() 方法创建另一个包含 int 元素的 3x3 数组 arr2 = np.arange(9).reshape((3,3)) print("
Array2...
", arr2) print("
数组类型...
", arr2.dtype) # 获取数组的维度 print("
数组维度...
",arr2.ndim) # 获取数组的形状 print("
我们的数组形状...
",arr2.shape) # 获取数组元素的数量 print("
数组中的元素...
",arr2.size) # 创建一个掩码数组 arr2 = ma.array(arr2) # 掩码数组 2 arr2[2, 1] = ma.masked arr2[2, 2] = ma.masked # 显示掩码数组 2 print("
Masked Array2...
",arr2) # 要返回两个掩码数组的点积,请使用 Python Numpy 中的 ma.dot() 方法 # "strict" 参数设置掩码数据是否传播 (True) 或设置为 0 (False) 进行计算 print("
Result of dot product...
",np.ma.dot(arr1, arr2, strict=True))

输出

Array1...
[[0 1 2]
[3 4 5]
[6 7 8]]

数组类型...
int64

数组维度...
2

我们的数组形状...
(3, 3)

数组中的元素...
9

掩码数组1...
[[0 -- 2]
[3 -- 5]
[6 7 8]]

Array2...
[[0 1 2]
[3 4 5]
[6 7 8]]

数组类型...
int64

数组维度...
2

我们的数组形状...
(3, 3)

数组中的元素...
9

掩码数组2...
[[0 1 2]
[3 4 5]
[6 -- --]]

点积结果...
[[-- -- --]
[-- -- --]
[69 -- --]]

相关文章