Первый месяц квартала данного месяца в Python

Question

В течение месяца в цифровой форме (например, 2 за февраль), как вы находите первый месяц своего квартала (например, 1 за январь)?

Я прочитал документацию модуля datetime и документацию Pandas о своих функциях datetime, что должно быть релевантным, но я не смог найти функцию, которая решает эту проблему.

По сути, то, что я пытаюсь понять, как я мог бы произвести функцию, как один ниже, что, учитывая месяц х, выводит число, соответствующее первый месяц х «s квартал.

>> first_month_quarter(5) 
4 

Answer 1

Это простая функция отображения, которая должна преобразовать:

1 2 3 4 5 6 7 8 9 10 11 12 
      | 
      V 
1 1 1 4 4 4 7 7 7 10 10 10 

Это может быть сделано в ряде способов с интегральным вычислением s, два из которых являются:

def firstMonthInQuarter(month): 
    return (month - 1) // 3 * 3 + 1 

и:

def firstMonthInQuarter(month): 
    return month - (month - 1) % 3 

Первый включает в себя целое деление месяца преобразуется в ноль на основе месяца, чтобы получить четверть с нуля, умножение, чтобы включить что обратно в месяц, основанный на нулевом значении (но месяц на начинается квартала), затем добавив один раз, чтобы сделать диапазон 1..12.

month -1 //3 *3 +1 
----- -- --- -- -- 
    1 0 0 0 1 
    2 1 0 0 1 
    3 2 0 0 1 
    4 3 1 3 4 
    5 4 1 3 4 
    6 5 1 3 4 
    7 6 2 6 7 
    8 7 2 6 7 
    9 8 2 6 7 
    10 9 3 9 10 
    11 10 3 9 10 
    12 11 3 9 10 

Второй раз вычитает положение в течение квартала (0, 1, 2) с самого месяца, чтобы получить начальный месяц.

month(a) -1 %3(b) a-b 
-------- -- ----- --- 
     1 0  0 1 
     2 1  1 1 
     3 2  2 1 
     4 3  0 4 
     5 4  1 4 
     6 5  2 4 
     7 6  0 7 
     8 7  1 7 
     9 8  2 7 
     10 9  0 10 
     11 10  1 10 
     12 11  2 10 

Answer 2

Вот ответ, предложенный TigerhawkT3. Возможно, самое скудное предложение до сих пор и, по-видимому, также было самым быстрым.

import math 

def first_month_quarter(month): 
    return int(math.ceil(month/3.)) * 3 - 2 

Например:

>> first_month_quarter(5) 
4 

Answer 3

def first_month(month): 
    return (month-1)//3*3+1 

for i in range(1,13): 
    print i, first_month(i) 

Answer 4

Это не так красиво, но если скорость важна простой список подстановки побоищ math:

def quarter(month, quarters=[None, 1, 1, 1, 4, 4, 4, 
          7, 7, 7, 10, 10, 10]): 
    """Return the first month of the quarter for a given month.""" 
    return quarters[month] 

timeit сравнение предполагает, что это примерно в два раза быстрее, чем математический подход TigerhawkT3.

---

тест сценарий:

import math 

def quarter(month, quarters=[None, 1, 1, 1, 4, 4, 4, 
          7, 7, 7, 10, 10, 10]): 
    """Return the first month of the quarter for a given month.""" 
    return quarters[month] 

def firstMonthInQuarter1(month): 
    return (month - 1) // 3 * 3 + 1 

def firstMonthInQuarter2(month): 
    return month - (month - 1) % 3 

def first_month_quarter(month): 
    return int(math.ceil(month/3.)) * 3 - 2 

if __name__ == '__main__': 
    from timeit import timeit 
    methods = ['quarter', 'firstMonthInQuarter1', 'firstMonthInQuarter2', 
       'first_month_quarter'] 
    setup = 'from __main__ import {}'.format(','.join(methods)) 
    results = {method: timeit('[{}(x) for x in range(1, 13)]'.format(method), 
           setup=setup) 
       for method in methods} 
    for method in methods: 
     print '{}:\t{}'.format(method, results[method]) 

Результаты:

quarter: 3.01457574242 
firstMonthInQuarter1: 4.51578357209 
firstMonthInQuarter2: 4.01768559763 
first_month_quarter: 8.08281871176 

Answer 5

упаковка таблицы поиска в 64-битного литерал:

def firstMonthOfQuarter(month): 
    return (0x000aaa7774441110L >> (month << 2)) & 15