Question

我试图按欧几里得距离对我的颜色（在CieLAB颜色空间中）进行排序。所以我正在使用以下代码，但是它重新排列了我的颜色，而不是进行排序。我是否需要指定其他轴或使用其他功能。如果我需要其他功能，可以建议哪个功能起作用？

a = np.array([(255,9,255), (0,0,0), (125,125,4)])
a.sort(axis=0)
print(a)

结果（请注意它是如何重新排列颜色的？）：

[[  0   0   0]
 [  4 125 125]
 [  9 255 255]]

应为：

[[  0   0   0]
 [  125 125 4]
 [  255 9 255]]

Answer 1

我希望我理解了这个问题，
也许您应该首先计算成对的距离，然后按这些距离排序。

类似的东西：

import numpy as np
from scipy.spatial.distance import cdist


def sort_by_eucledian_distance(a):
    dist = cdist(a, a)[:, 0] # calculate distances
    dist = sorted(zip(dist, np.arange(len(dist)))) #add indexes and sort
    idxs = [v[1] for v in dist] # get the new, sorted indexes
    return a[idxs]

a = np.array([(255,9,255), (0,0,0), (125,125,4)])
b = sort_by_eucledian_distance(a)
print(b)

将打印

array([[  0.,   0.,   0.],          
       [125.,   9.,   4.],          
       [255., 125., 255.]])

Answer 2

如果我正确理解了您的问题，您想按照欧几里得范数排序，对吗？

类似

a[np.linalg.norm(a, axis=1).argsort()]

或使用np.einsum（按a dot a行排序）

a[np.einsum("ij,ij->i", a, a).argsort()]

按欧几里得距离对颜色列表进行排序

2 个答案: