python - (P, Q) 和 (R,) 之间的曼哈顿距离

Question

我正在研究曼哈顿距离。它适用于简单的 for 循环。但我试图避免这种 for 循环。

import numpy as np
import random
A = np.random.randint(5, size=(10, 5))
B = [1, 3, 5, 2, 4]
for i in range(10):
    dist = sum(abs(A[i]-B))
    print("Distances: ", dist)

还有比这更好的方法吗？例如使用高级索引..谢谢您的指导。

标签： pythonarrayspython-3.xnumpy

Answer 1

纯的numpy

您可以在 numpy 中执行此操作：

>>> np.sum(np.abs(A-B), axis=1)
array([10,  6,  9,  9,  7,  7,  9,  8, 14,  8])

将此与循环的输出进行比较：

Distances:  10
Distances:  6
Distances:  9
Distances:  9
Distances:  7
Distances:  7
Distances:  9
Distances:  8
Distances:  14
Distances:  8

选择：scipy

如果你愿意，你也可以使用scipy（虽然我个人更喜欢这种numpy方法）：

from scipy.spatial.distance import cdist

>>> cdist(A,np.array(B).reshape(1,-1), metric='cityblock')
array([[10.],
       [ 6.],
       [ 9.],
       [ 9.],
       [ 7.],
       [ 7.],
       [ 9.],
       [ 8.],
       [14.],
       [ 8.]])