python-3.x - 根据多列设置自第一次出现以来的天数

Question

我有一个具有以下结构的 pandas 数据集:

Date          datetime64[ns]
Events                 int64
Location              object
Day                  float64

我使用以下代码来获取位置“A”第一次出现的日期:

start_date = df[df['Location'] == 'A'][df.Events != 0].iat[0,0]

我现在想要用自 start_date 以来的天数更新 start_date 之后的所有记录，其中 Day = df.Date - start_date。

我尝试了这段代码:

df.loc[df.Location == country, 'Day'] = (df.Date - start_date).days

但是，该代码返回错误:

AttributeError: 'Series' object has no attribute 'days'

问题似乎是代码将 df.Date 识别为对象而不是日期时间。有人对导致此问题的原因有任何想法吗？

Answer 1

试试吧，需要添加 .dt accessor .

df.loc[df.Location == country, 'Day'] = (df.Date - start_date).dt.days