python3中的自定义比较函数

Question

我想按点到原点 (0,0) 的距离以升序对 2D 坐标系中的点进行排序。我找到了this 并尝试了一些方法，但仍然无法获得想要的结果。

这是我的代码：

from functools import cmp_to_key

def my_comp(point1, point2):
    return point1[0]*point1[0] + point1[1]*point1[1] < point2[0]*point2[0] + point2[1]*point2[1]

points = [ [3.1, 4.1], [0.9, 0.8], [1.0, 1.0] ]
sorted(points, key=cmp_to_key(my_comp))
print(points)

结果：

[[3.1, 4.1], [0.9, 0.8], [1.0, 1.0]]

预期：

[[0.9, 0.8], [1.0, 1.0], [3.1, 4.1]]

Answer 1

看起来你在这里经历了一些额外的障碍。你有一个已经是完美键的数量。您不仅可以使用它，还可以定义一个比较器，该比较器为每对对象重新计算和比较其中两个量，然后将该比较器转换回一个键。

这似乎非常低效。为什么不直接定义一个简单的按键函数直接使用呢？

def distance_from_origin_squared(point):
    return point[0]**2 + point[1]**2
points = sorted(points, key=distance_from_origin_squared)

请记住，sorted 不会就地运行，因此您必须将其存放在某个地方。如果可以，请替换原始名称。如果您想要就地排序，请改用list.sort：

points.sort(key=distance_from_origin_squared)

从实用的角度来看，与比较器相比，键具有许多优势。一方面，一个关键函数只需要每个元素调用一次，而比较器调用O(nlog(n)) 次。当然，键会比较O(nlog(n)) 次，但比较通常要简单得多，因为感兴趣的值是预先计算的。

来自 C 的主要困难（在我的例子中）是理解键不必是单个整数的想法，如示例中所示。键可以是明确定义<（或>）运算符的任何类型。字符串、列表和元组的字典比较可以很容易地将复杂的链式比较器归结为只是一个短序列值的键。

Answer 2

1) 您的 my_cmp() 函数应该返回三个值之一（+、- 或 0，具体取决于比较），但您只返回两个值（True 和 False）。

2) 你忽略了sorted() 的返回值。 sorted() 不修改它的参数，它返回它的排序副本。

3) 不要使用 cmp 函数。它们很难描述，也很难实施。而是使用键功能。

怎么样：

def my_key(point1):
    return point1[0]*point1[0] + point1[1]*point1[1]

points = [ [3.1, 4.1], [0.9, 0.8], [1.0, 1.0] ]
points = sorted(points, key=my_key)
print(points)

assert points == [ [0.9, 0.8], [1.0, 1.0], [3.1, 4.1] ]

Answer 3

这是你想要的吗？

sorted_points = sorted(points, key=lambda p: p[0]*p[0] + p[1]*p[1])

在您的原始代码中要注意的另一件事是 sorted 不会对列表进行适当的排序，它会创建一个已排序的新列表。所以当你这样做时

points = [3,2,1]
sorted(points)
print(points)

您的代码似乎没有对任何内容进行排序，因为您打印的是原始列表而不是新创建的排序列表。您可以像这样就地执行排序。

points = [3,2,1]
points.sort()
print(points)

Answer 4

您混淆了 cmp 和 key 参数。 cmp 函数接受两个参数并返回 -1、0 或 1。 key 函数返回将用于对列表进行排序的代理值。 cmp 在 Python 3 中作为 sorted 的参数被删除。