日付の間隔を数値型に変換

Question

前提・実現したいこと

PandasのDataFrameにおいて、datetimeで作成した日付の間隔を使って、相関分析などの数値演算をしたい。

発生している問題・エラーメッセージ

次のようなエラーが出る。

TypeError: ufunc multiply cannot use operands with types dtype('<m8[ns]') and dtype('<m8[ns]')

該当のソースコード

# ライブラリのインポート
import pandas as pd
from datetime import datetime
from scipy.stats import pearsonr

# データフレームの作成
df = pd.DataFrame({
    'date':[datetime(2018,6,1),datetime(2018,6,2),datetime(2018,6,3)],
    'value':[1.2, 3.5, 4.7]
})

# 日数の間隔（days）を追加
today = datetime(2018, 6, 1)
df['days'] = df['date'] - today

# 相関係数の確認
print(pearsonr(df.days, df.value))

試したこと

列daysが数値型ではないからエラーが出るのだと考え、次のコードを書いて型変換をしようとしました。

# 数値型への変換
df['days'] = df['days'].astype('int')

しかし、以下のようなエラーが出ました。

TypeError: cannot astype a timedelta from [timedelta64[ns]] to [int32]

Answer 1

df['days'] = df['date'] - today

の箇所を

df['days'] = (df['date'] - today).dt.days

とすることで、日数（int型）に変換できるかと思います。

Comments

[naozoo]

無事解決できました。
ありがとうございました！