Python按绝对值排序列表时如何实现同绝对值正数优先级更高
整数列表按绝对值排序(同绝对值正数优先)的优雅实现
你之前用加0.1偏移的写法属于hack式实现,存在浮点数精度冲突的隐患,可读性也差,利用Python排序key支持元组逐位比较的特性,就可以写出严谨简洁的实现:
实现思路
排序规则拆分成两个优先级,按顺序比较即可:
- 第一排序维度:按数值的绝对值从小到大排序,作为元组key的第一个元素
- 第二排序维度:绝对值相等时,正数排在负数前面。由于Python排序默认升序,只需要让正数对应的第二个key值小于负数对应的key值即可。
代码实现
利用Python布尔值是整数子类的特性(False等价于0,True等价于1),可以写出非常简洁的代码:
lst = [1, -3, 3, -3, 12, 10] res = sorted(lst, key=lambda x: (abs(x), x < 0)) print(res) # 输出:[1, 3, -3, -3, 10, 12]
逻辑说明
- 对任意正数x:
x < 0返回False(对应数值0) - 对任意负数x:
x < 0返回True(对应数值1)
同绝对值的正数和负数比较时,第二个key值0 < 1,因此正数会排在负数前面,完全符合需求。
这种写法没有任何魔法偏移值,不存在精度问题,代码语义清晰,也兼容包含0、浮点数的列表场景。
内容的提问来源于stack exchange,提问作者Mortz
相关产品推荐
相关产品推荐

