按歐幾里得距離對顏色列表進行排序

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()]