我试图将一个二维数组与两个一维数组的乘积(联合概率密度与其个体概率密度的乘积)进行比较,以确定变量 x 和 y 是否独立,其中独立性为 { {1}}。
假设我称二维数组为 ρ(x,y)=ρ(x)*ρ(y)
,一维列表为 h
和 n
。我将如何遍历 m
以检查它的元素是否等效于 h
?
答案 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))))