将二维数组的元素与两个一维 arrays 的乘积进行比较？

Question

我正在尝试将 2D 数组与两个 1D arrays 的乘积（联合概率密度与其个体概率密度的乘积）进行比较，以确定变量 x 和 y 是否独立，其中独立性为ρ(x,y)=ρ(x)*ρ(y) 。

假设我调用了二维数组h ，而一维数组n和m 。 我将如何 go 关于迭代h以检查它的元素是否等于n*m ？

Answer 1

要测试完全相等，只需使用np.all()

import numpy as np

m = np.random.rand(10)
n = np.random.rand(20)
h = m.reshape(1, -1) * n.reshape(-1, 1)

print(np.all(h == m.reshape(1, -1) * n.reshape(-1, 1))) # True

要测试数字是否都接近，您可以使用：

print(np.all(np.isclose(h, m.reshape(1, -1) * n.reshape(-1, 1))))