Pandas: como calcular o timedelta em meses

Você pode usar a seguinte função para calcular um delta de tempo em meses entre duas colunas de um DataFrame do pandas:

 def month_diff(x, y):
    end = x. dt . to_period (' M '). view (dtype=' int64 ')
    start = y. dt . to_period (' M '). view (dtype=' int64 ')
    return end-start

O exemplo a seguir mostra como usar esta função na prática.

Exemplo: Calcular Timedelta em meses no Pandas

Suponha que temos o seguinte DataFrame do pandas:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' event ': ['A', 'B', 'C'],
                   ' start_date ': ['20210101', '20210201', '20210401'],
                   ' end_date ': ['20210608', '20210209', '20210801'] })

#convert start date and end date columns to datetime
df[' start_date '] = pd. to_datetime (df[' start_date '])
df[' end_date '] = pd. to_datetime (df[' end_date '])

#view DataFrame
print (df)

  event start_date end_date
0 A 2021-01-01 2021-06-08
1 B 2021-02-01 2021-02-09
2 C 2021-04-01 2021-08-01

Agora suponha que queiramos calcular o delta de tempo (em meses) entre as colunas data_inicial e data_final .

Para fazer isso, primeiro definiremos a seguinte função:

 #define function to calculate time delta in months between two columns
def month_diff(x, y):
    end = x. dt . to_period (' M '). view (dtype=' int64 ')
    start = y. dt . to_period (' M '). view (dtype=' int64 ')
    return end-start

A seguir, usaremos esta função para calcular o delta de tempo em meses entre as colunas start_date e end_date :

 #calculate month difference between start date and end date columns
df[' month_difference '] = month_diff(df. end_date , df. start_date )

#view updated DataFrame
df

    event start_date end_date month_difference
0 A 2021-01-01 2021-06-08 5
1 B 2021-02-01 2021-02-09 0
2 C 2021-04-01 2021-08-01 4

A coluna Month_difference exibe o delta de tempo (em meses) entre as colunas start_date e end_date .

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras operações comuns em pandas:

Como converter colunas para DateTime no Pandas
Como converter DateTime em data no Pandas
Como extrair o mês da data no Pandas