pandasで2つの日数の差を列に加えたい

Question

pandasのdataframeで同じ行の2つの年月日情報(int型)から差の日数を求めたいです。
しかし、以下のようなコードではシリーズ型をintに変換できないと言うことでエラーになってしまいます。

df['days'] = (datetime.date(year = df['year'] ,month = df['month'], day = df['day'])- datetime.date(year = df['year2'], month = df['month2'], day = df['day2'])).days

TypeError: cannot convert the series to <class 'int'>

このようにデータフレームの列にある年月日情報から日数の差を求める方法があればご教授願いたいです。
for文で考えたのですが行が数万あり、実行にかかる時間が膨大となったので早い方法だと助かります。
拙い質問ですが、どうぞよろしくお願いします。

Answer 1

pandas.to_datetime()に、year, month, day等のカラム名のデータフレームを渡すと、datetime型に変換してくれます。year, month, day等をキーにもつdictでもいいです。
(リファレンスのExampleのところ参照)

import pandas as pd

df = pd.DataFrame({
    'year': [2021, 2021, 2021],
    'month': [6, 5, 4],
    'day': [20, 1, 3],
    'year2': [2021, 2021, 2021],
    'month2': [1, 2, 3],
    'day2': [5, 10, 15],
})

df['days'] = (
    pd.to_datetime(df[['year', 'month', 'day']])
    - pd.to_datetime({'year': df['year2'], 'month': df['month2'], 'day': df['day2']})
).dt.days

Comments

[Mathroat]

ありがとうございます！