当前函数的哪一个实现更像python？

Question

current_quarter函数的哪个实现更具python风格，为什么？你知道吗

第一个版本似乎更容易扩展。然而，这是一种你不会扩展的函数，因为四分之一的概念不会改变。你知道吗

在第二个版本中，更容易掌握力学。你知道吗

def current_quarter(input_date):
    months_baskets = [(1,2,3), (4,5,6), (7,8,9), (10,11,12)]
    quarters = (1,2,3,4)
    for idx, basket in enumerate(months_baskets):
        if input_date.month in basket:
            return quarters[idx]
            break

def current_quarter(input_date):
    if input_date.month in (1, 2, 3):
        return 1
    elif input_date.month in (4, 5, 6):
        return 2
    elif input_date.month in (7, 8, 9):
        return 3
    elif input_date.month in (10, 11, 12):
        return 4

Answer 1

我不确定我是否得到了Martjin的建议，所以我冒昧地提出我的解决方案

>>> def current_quarter(input_date):
        return (input_date.month - 1) // 3 + 1
>>>
>>> from datetime import datetime
>>> [(m, current_quarter(datetime(2014, m, 1))) for m in range(1,13)]
[(1, 1), (2, 1), (3, 1), (4, 2), (5, 2), (6, 2), (7, 3), (8, 3), (9, 3), (10, 4), (11, 4), (12, 4)]