Python: как получить разницу между двумя месяцами от целых чисел?

Question

У меня есть фрейм данных df, например:

df
    A    NUM_YYYYMM
0   a     201605
1   b     201602
2   c     201603
3   d     201601

, где type(df['NUM_YYYYMM']) возвращает int.Я хочу вычислить разницу в месяцах между t0=201612 и столбцом df['NUM_YYYYMM'].Итак:

df
    A    NUM_YYYYMM   deltaT
0   a     201605        7
1   b     201602       10
2   c     201603        9
3   d     201601       11

Answer 1

Вы можете определить свое собственное вычитание, используя целочисленное и модульное деление, учитывая ваш стандартный формат ГГГГММ

def my_subtract(x, t0):
    return (t0//100 - x//100)*12 + (t0%100 - x%100)

df['deltaT'] = my_subtract(df.NUM_YYYYMM, 201612)

Вывод df:

   A  NUM_YYYYMM  deltaT
0  a      201605       7
1  b      201602      10
2  c      201603       9
3  d      201601      11

Answer 2

Преобразовать столбец to_datetime, а затем в месячные периоды на to_period, который вычитается из Period из t0:

t0 = '201612'
t = pd.to_datetime(t0, format='%Y%m').to_period('m')

df['deltaT'] = t - pd.to_datetime(df['NUM_YYYYMM'], format='%Y%m').dt.to_period('m')
print (df)
   A  NUM_YYYYMM deltaT
0  a      201605      7
1  b      201602     10
2  c      201603      9
3  d      201601     11

Если возможно, измените формат t0:

t = pd.Period('2016-12')
df['deltaT'] = t - pd.to_datetime(df['NUM_YYYYMM'], format='%Y%m').dt.to_period('m')