在 Numpy 中返回两个掩码数组的点积
numpyserver side programmingprogramming更新于 2025/9/3 6:37:17
要返回两个掩码数组的点积，请使用 Python Numpy 中的 ma.dot() 方法。此函数相当于将掩码值考虑在内的 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)
获取数组的维度 −
print("
数组维度...
",arr1.ndim)
获取数组的形状 −
print("
我们的数组形状...
",arr1.shape)
获取数组元素的数量 −
print("
数组中的元素...
",arr1.size)
创建掩码数组 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("
数组2...
", 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() 方法 −
print("
点积结果...
",np.ma.dot(arr1, arr2))
示例
import numpy as np
import numpy.ma as ma
# 数组 1
# 使用 numpy.arange() 方法创建一个包含 int 元素的 3x3 数组
arr1 = np.arange(9).reshape((3,3))
print("数组 1...
", 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("
掩码数组 2...
",arr2)
# 要返回两个掩码数组的点积，请使用 Python Numpy 中的 ma.dot() 方法
print("
点积结果...
",np.ma.dot(arr1, arr2))
输出
Array1... [[0 1 2] [3 4 5] [6 7 8]] 数组类型... int64 数组维度... 2 我们的数组形状... (3, 3) 数组中的元素... 9 掩码数组 1... [[0 -- 2] [3 -- 5] [6 7 8]] 数组 2... [[0 1 2] [3 4 5] [6 7 8]] 数组类型... int64 数组维度... 2 我们的数组形状... (3, 3) 数组中的元素... 9 掩码数组2... [[0 1 2] [3 4 5] [6 -- --]] 点积结果... [[12 0 0] [30 3 6] [69 34 47]]
