2015-09-28 58 views
1

Alphabet Data List如何使用Python3中的两个条件对对象进行排序?

  • 字母表中的列表表示对象。
  • 每个字母都有一个中心坐标(x,y)。
  • 该数字表示其列表中的索引。

我想按照[A,B,C,D]的顺序排列项目。所以...

if abs(A.y - B.y) < threshold: 
    # sort by x coordinate 
else: 
    # sort by y coordinate 

我可以通过手动检查对象并直接交换它们的位置来做到这一点。

但我该怎么用Python3​​函数做这个?

回答

4

写一个比较函数,然后用functools.cmp_to_key把它转换成一个关键功能:

# Given a threshold, return a function suitable for 
# use by old cmp argument 
def comparator(threshold): 
    def compare(A, B): 
     if abs(A.y - B.y) < threshold: 
      return cmp(A.x, B.x) 
     else: 
      return cmp(A.y, B.y) 
    return compare 

from functools import cmp_to_key 
my_cmp = comparator(0.6) # Or whatever threshold you need 
sorted_list = sorted(my_list, key=cmp_to_key(my_cmp)) 
+0

打我几秒:-) –

-1
sorted(your_list, key=lambda p: (p.x, p.y)) 
+0

这没有考虑到'y'坐标差异的阈值。 – chepner

3

按照python3文档:Sorting How To你会定义一个复杂的比较功能,然后用functools。 cmp_to_key将其转换为一个关键的功能:

这应该工作:

import functools 
def comp_func(A,B): 
    if abs(A.y - B.y) < threshold: 
     return A.y - B.y # Sort by y co-ord 
    else: 
     return A.x - B.x # Sort by x co-ord 

.... 
sorted(data, key=functools.cmp_to_key(comp_func)) 
相关问题