python - 按升序对月份名称字符串列表进行排序

Question

有人可以指导如何对月份列表进行排序吗？

INPUT = ['August', 'September', 'October', 'November', 'December', 'January']

OUTPUT= ['January','August', 'September', 'October', 'November', 'December']

Answer 1

calendar.month_name是一个可以按正确顺序转换为月份列表的对象。假设您的输入字符串格式正确，您可以按此列表中每个字符串的索引排序:

>>> from calendar import month_name
>>> month_lookup = list(month_name)
>>> months = ['August', 'September', 'October', 'November', 'December', 'January']
>>> sorted(months, key=month_lookup.index)
['January', 'August', 'September', 'October', 'November', 'December']

您还可以使用 datetime.strptime和匹配完整月份名称的 "%B" 格式字符串。

>>> from datetime import datetime
>>> months = ['August', 'September', 'October', 'November', 'December', 'January']
>>> sorted(months, key=lambda m: datetime.strptime(m, "%B"))
['January', 'August', 'September', 'October', 'November', 'December']

请注意，这两种解决方案都会在包含无效月份字符串的列表上引发 ValueError。日期时间版本给出了更清晰的信息。