如何在沒有循環的情況下計算一維和二維數組之間的漢明距離

Question

A 是形狀為 100 的一維數組，B 是形狀為 (50000, 100) 的二維數組。 我想計算 A 和 B 之間的漢明距離，並得到一個形狀為 50000 的數組 X。

我可以用一個循環來做到這一點：

for i in range(50000):
    X[i] = np.count_nonzero(A != B[j,:])

我想知道我可以跳過循環還是做一些事情來使它更快？

Answer 1

您可以直接將A和B與A != B進行比較，由於A和B的維數不同，這將廣播，然后您可以使用每行np.count_nonzero和axis=1 ：

np.count_nonzero(A != B, axis=1)

---

A = np.array([1,2])
B = np.array([[1,2],[3,2],[1,3],[2,4]])

np.count_nonzero(A != B, axis=1)
# array([0, 1, 1, 2])