pythonのdatetime型からtimedelta型への変換

Question

datetime型のpandasデータフレームを、timedelta型に変換したいです。

df
   time
0 06:11:00
1 06:13:00
2 06:18:00
3 06:40:00

type(df['time'][0])
# datetime.time
df['time'][0]
# datetime.time(6, 11)

このデータフレームをtimedelta型にしたいのですがうまくいきません。
train['time'] = pd.to_timedelta(train['time'], unit='min')
などと書いてみますが、
Invalid type for timedelta scalar: <class 'datetime.time'>
と、エラーが出ます。
どのようなコード書くと変換できるのでしょうか。

Answer 1

以下のようにtimeから秒数を計算してto_timedeltaすればよいです。

from datetime import time
import pandas as pd

df = pd.DataFrame({'time':[time(6,11), time(6,13)]})
print(type(df['time'][0])) # <class 'datetime.time'>

sec = df['time'].apply(lambda t:(t.hour * 60 + t.minute) * 60 + t.second)
df['td'] = pd.to_timedelta( sec, unit='second')
print(type(df['td'][0])) # <class 'pandas._libs.tslibs.timedeltas.Timedelta'>
print(df)
#       time              td
#0  06:11:00 0 days 06:11:00
#1  06:13:00 0 days 06:13:00

Comments

[fu_3823]

ありがとうございました。
とても参考になりました。