Первый месяц квартала данного месяца в 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
