按月python排序元组列表

时间:2018-04-11 19:35:43

标签: python sorting

我有元组中的假期列表:

holidays = [('Easter', ('Apr', 1), 0),
            ('Earth Day', ('Apr', 22), 0),
            ('Pearl Harbor Rememberance Day', ('Dec', 7), 0),
            ('Christmas Day', ('Dec', 25), 50),
            ('Groundhog Day', ('Feb', 2), 0)]

以及对它们进行排序的功能:

def holiday1(holidays : [(str,(str,int),int)]) -> [(str,(str,int),int)]:
    return sorted(holidays, key=lambda x:x[1])

我按字母顺序按月分类,但我想按月份的位置对它们进行排序。 (例如Jan = 1,Feb = 2等)

我已经创建了几个月的词典:

months = {'Jan': 1, 'Feb': 2, 'Mar': 3, 'Apr': 4, 'May': 5, 'Jun': 6, 'Jul': 7, 'Aug': 8, 'Sep': 9, 'Oct': 10, 'Nov': 11, 'Dec': 12}

我不确定如何使用dict'月份'对元组列表进行排序,以便得出结果:

holidays = [('Groundhog Day', ('Feb', 2), 0),
            ('Easter', ('Apr', 1), 0),
            ('Earth Day', ('Apr', 22), 0),
            ('Pearl Harbor Rememberance Day', ('Dec', 7), 0),
            ('Christmas Day', ('Dec', 25), 50)]

3 个答案:

答案 0 :(得分:5)

我们排序的元素是3元组,其中第二个元素又是元组。我们对月份感兴趣,它是元组的第二个元素的第一个元素,因此我们可以使用x[1][0]访问它。但是我们当然不希望按字母顺序排序,而是按月的顺序排序,因此我们可以使用months[x[1][0]]执行查找。因此,我们只需将考虑在内就可以对其进行排序:

# only taking month into account
return sorted(holidays, key=lambda x: months[x[1][0]])

但我的猜测是,如果是 tie ,你想要考虑这一天。关于元组的好处是,如果你比较它们,Python将首先包含两个元组的第一个元素。如果是平局,它将比较两个元组的第二个元素等。我们可以通过让key返回2元组来使用它:首先是月份的索引,然后是日期:

# taking month and day into account
return sorted(holidays, key=lambda x: (months[x[1][0]], x[1][1]))

编辑:如果你想在月份和日期相等的情况下按名称对假期进行排序,我们可以构建3元组:

# taking month, day and name into account
return sorted(holidays, key=lambda x: (months[x[1][0]], x[1][1], x[0]))

答案 1 :(得分:2)

使用键功能对要排序的对象进行排名。

def by_month(holiday):
    """sort by month, then day, then name of holiday"""
    holiday_name = holiday[0]
    month, day = holiday[1]
    month_value = months[month]
    return month_value, day, holiday_name

sorted(holidays, key=by_month)

输出为

[('Groundhog Day', ('Feb', 2), 0),
 ('Easter', ('Apr', 1), 0),
 ('Earth Day', ('Apr', 22), 0),
 ('Pearl Harbor Rememberance Day', ('Dec', 7), 0),
 ('Christmas Day', ('Dec', 25), 50)]

答案 2 :(得分:0)

不要使用ad-hoc数据结构和值来表示假期,而是定义其值正确行为的类型。

首先,创建几个月的专用类型,而不是仅使用常规字符串。

from enum import IntEnum, auto, unique, EnumMeta

Month = IntEnum('Month',
        "Jan Feb Mar Apr May Jun Jul Aug Sep Oct Nov Dec".split())

接下来,定义一个表示假日的类,而不是使用普通元组。

from functools import total_ordering

@total_ordering
class Holiday:
    def __init__(self, name, month, day, x):
        self.name = name
        self.month = month
        self.day = day
        self.x = x
        self.sort_key = (self.month, self.day, self.name)

    def __lt__(self, other):
        return self.sort_key < other.sort_key

    def __eq__(self, other):
        return self.sort_key == other.sort_key and self.x == other.x

    def __str__(self):
        return "{} ({} {})".format(self.name, self.month._name_, self.day)

现在您的假期可以排序而无需任何进一步的努力:

holidays = [
    Holiday('Easter', Month.Apr, 1, 0),
    Holiday('Earth Day', Month.Apr, 22, 0),
    Holiday('Pearl Harbor Rememberance Day', Month.Dec, 7, 0),
    Holiday('Christmas Day', Month.Dec, 25, 50),
    Holiday('Groundhog Day', Month.Feb, 2, 0)
    ]

for h in sorted(holidays):
    print(h)

输出

Groundhog Day (Feb 2)
Easter (Apr 1)
Earth Day (Apr 22)
Pearl Harbor Rememberance Day (Dec 7)
Christmas Day (Dec 25)