將二維數組的元素與兩個一維 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))))