对一个列表进行排序以使两个列表具有正确的顺序对应关系

Question

我有两个列表，例如

coords = [2, 0, 1, 4, 3]
value = [1, 9, 3, 3, 0]

其中第一个是一系列坐标，第二个是与坐标对应的一系列值，例如坐标'2' 对应于值'1'，坐标'0' 给出值'9'。

现在，我想对coords进行排序但是保持value的顺序不变，这样最小的coords元素对应于value中的最小元素，以此类推在。期望的输出是：

coords_new = [1, 4, 2, 3, 0]
value = [1, 9, 3, 3, 0] # unchanged

在哪里'0' -> '0', '1' -> '1', '2' -> '3', '3' -> '3', '4' -> '9'。 有什么想法吗？您可以返回 coords_new，或将 coords 重新排序的索引作为答案。

编辑： 如果可能的话，我希望我们可以返回对原始coords 重新排序的索引，即返回idx 使得coords[idx] = coords_new。

非常感谢！

志豪

Answer 1

这里是使用 argsort 的一个半解决方案。 kind='mergesort' kwd 参数仅在您需要稳定排序时才需要。在您的示例中，不稳定的排序也可能产生coords_new == [1, 4, 3, 2, 0]。如果这不是问题，您可以省略 kwd 参数并允许 numpy 使用更快的排序算法。

import numpy as np

coords = [2, 0, 1, 4, 3]
value = [1, 9, 3, 3, 0]

coords, value = map(np.asanyarray, (coords, value))

vidx = value.argsort(kind='mergesort') # mergesort is stable, i.e. it  
                                       # preserves the order of equal elements

# direct method:
coords_new = np.empty_like(coords)
coords_new[vidx] = np.sort(coords)

# method yielding idx
idx = np.empty_like(vidx)
idx[vidx] = coords.argsort(kind='mergesort') 

第二种方法产生idx，这样coords_new == coords[idx]。

Answer 2

我假设你想要一个 numpy 的答案，因为你已经标记了 numpy：

>>> x = np.argsort(value)
>>> x[x]
array([1, 4, 2, 3, 0])

Answer 3

另一种方法是先创建对象之间的映射，然后将此映射与索引结合使用：

coords = [2, 0, 1, 4, 3]
value = [1, 9, 3, 3, 0]

table = {k: v for k, v in zip(sorted(coords), sorted(value))}
print(table)
print(sorted(coords, key=lambda e: value.index(table[e])))

输出

{0: 0, 1: 1, 2: 3, 3: 3, 4: 9}
[1, 4, 2, 3, 0]

注意

此方法假定coords 仅包含唯一值。对于一般情况，您可以生成映射的对 (c, v) 按值中 v 的索引值排序：

pairs = [(k, v) for k, v in zip(sorted(coords), sorted(value))]
result = [k for k, _ in sorted(pairs, key=lambda e: value.index(e[1]))]

print(result)

输出

[1, 4, 2, 3, 0]