เพิ่มคอลัมน์ที่มีจำนวนวันระหว่างวันที่ใน DataFrame แพนด้า

ฉันต้องการลบวันที่ใน 'A' จากวันที่ใน 'B' และเพิ่มคอลัมน์ใหม่ที่มีความแตกต่าง

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

ฉันได้ลองทำสิ่งต่อไปนี้แล้ว แต่ได้รับข้อผิดพลาดเมื่อฉันพยายามรวมสิ่งนี้ในการวนซ้ำ ...

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

ฉันควรทำอย่างไรดี?

สมมติว่าคอลัมน์เหล่านี้เป็นคอลัมน์วันที่และเวลา (หากไม่ใช้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

หมายเหตุ: ตรวจสอบให้แน่ใจว่าคุณใช้แพนด้าตัวใหม่ (เช่น 0.13.1) ซึ่งอาจใช้ไม่ได้กับเวอร์ชันเก่า

หากต้องการลบองค์ประกอบข้อความ "days" คุณยังสามารถใช้ตัวเข้าถึง dt () สำหรับชุดข้อมูลได้: https://pandas.pydata.org/pandas-docs/stable/generated/pandas.Series.dt.html

ดังนั้น,

df[['A','B']] = df[['A','B']].apply(pd.to_datetime) #if conversion required
df['C'] = (df['B'] - df['A']).dt.days

ซึ่งส่งคืน:

             A          B   C
one 2014-01-01 2014-02-28  58
two 2014-02-03 2014-03-01  26

ความเข้าใจในรายการเป็นทางออกที่ดีที่สุดสำหรับวิธี Pythonic ที่สุด (และเร็วที่สุด) ในการทำสิ่งนี้:

[int(i.days) for i in (df.B - df.A)]

1. ฉันจะส่งคืน timedelta (เช่น '-58 วัน')
2. i.days จะคืนค่านี้เป็นค่าจำนวนเต็มยาว (เช่น -58L)
3. int (i.days) จะให้ -58 ที่คุณต้องการ

หากคอลัมน์ของคุณไม่ได้อยู่ในรูปแบบวันที่และเวลา ไวยากรณ์ที่สั้นกว่าจะเป็น:df.A = pd.to_datetime(df.A)

แล้วสิ่งนี้:

times['days_since'] = max(list(df.index.values))  
times['days_since'] = times['days_since'] - times['months']  
times