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

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

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

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

>> first_month_quarter(5)
4

Ответ 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

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

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

Ответ 2

Это не очень красиво, но если скорость важна, простейшие поисковые запросы списка 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]

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

Тест script:

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

Ответ 3

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

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

Ответ 4

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

import math

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

Например:

>> first_month_quarter(5)
4

Ответ 5

Упаковка таблицы поиска в 64-разрядный литерал:

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