1. ホーム
  2. python

[解決済み] DataFrameに日付間の日数カラムを追加する pandas

2022-07-18 19:17:58

質問

A'の日付から'B'の日付を引き、その差分を新しい列として追加したいのですが、可能でしょうか?

df
          A        B
one 2014-01-01  2014-02-28 
two 2014-02-03  2014-03-01

以下のように試してみましたが、これをforループに含めようとするとエラーになります...。

import datetime
date1=df['A'][0]
date2=df['B'][0]
mdate1 = datetime.datetime.strptime(date1, "%Y-%m-%d").date()
rdate1 = datetime.datetime.strptime(date2, "%Y-%m-%d").date()
delta =  (mdate1 - rdate1).days
print delta

どうすればいいのでしょうか?

どのように解決するのですか?

これらがdatetimeカラムであったと仮定して(もし適用されていない場合 to_datetime でない場合)、それらを引き算することができます。

df['A'] = pd.to_datetime(df['A'])
df['B'] = pd.to_datetime(df['B'])

In [11]: df.dtypes  # if already datetime64 you don't need to use to_datetime
Out[11]:
A    datetime64[ns]
B    datetime64[ns]
dtype: object

In [12]: df['A'] - df['B']
Out[12]:
one   -58 days
two   -26 days
dtype: timedelta64[ns]

In [13]: df['C'] = df['A'] - df['B']

In [14]: df
Out[14]:
             A          B        C
one 2014-01-01 2014-02-28 -58 days
two 2014-02-03 2014-03-01 -26 days

注意: pandasの新しいバージョン(例えば0.13.1)を使用していることを確認してください。