我有元组中的假期列表:
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)]
答案 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)