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

时间:2021-02-25 21:30:04

标签: python arrays statistics probability-density

我试图将一个二维数组与两个一维数组的乘积(联合概率密度与其个体概率密度的乘积)进行比较,以确定变量 x 和 y 是否独立,其中独立性为 { {1}}。

假设我称二维数组为 ρ(x,y)=ρ(x)*ρ(y),一维列表为 hn。我将如何遍历 m 以检查它的元素是否等效于 h

1 个答案:

答案 0 :(得分: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))))