Python pandas .loc多个条件并修改部分日期

时间:2016-04-05 09:44:56

标签: python datetime pandas multiple-conditions

我有一个DataFrame,其中包含两个格式如下的日期列:

cap['DateCollecte'] = pd.to_datetime(cap['Date de collecte']+' '+cap['Heure de collecte'],format='%d/%m/%Y %H:%M:%S',errors='coerce')
cap['DatePose'] = pd.to_datetime(cap['Date de pose']+' '+cap['Heure de pose'],format='%d/%m/%Y %H:%M:%S',errors='coerce')

我知道其中一些有错误:DatePose的年份是2010年,DateCollecte的年份是2011年.DatePose的年份也应该是2011年。 我想做的事情的想法或多或少:

cap.loc[(cap.DateCollecte.dt.year == 2011) & (cap.DatePose.dt.year == 2010), cap.DatePose.dt.year] = 2011

我认为棘手的部分是我只想改变约会的年份部分。有没有办法做到这一点?

在最糟糕的情况下,因为它只涉及3行不同日期(但同年问题),我可以手工完成。

1 个答案:

答案 0 :(得分:1)

您可以使用面具并添加DateOffset,例如:

In [43]:
df=pd.DataFrame({'a':pd.date_range(dt.datetime(2011,1,1), dt.datetime(2012,1,1), freq='M'), 'b':pd.date_range(dt.datetime(2010,6,1), dt.datetime(2011,6,1), freq='M')})
df

Out[43]:
            a          b
0  2011-01-31 2010-06-30
1  2011-02-28 2010-07-31
2  2011-03-31 2010-08-31
3  2011-04-30 2010-09-30
4  2011-05-31 2010-10-31
5  2011-06-30 2010-11-30
6  2011-07-31 2010-12-31
7  2011-08-31 2011-01-31
8  2011-09-30 2011-02-28
9  2011-10-31 2011-03-31
10 2011-11-30 2011-04-30
11 2011-12-31 2011-05-31

In [65]:
df.loc[(df['a'].dt.year == 2011) & (df['b'].dt.year == 2010), 'b'] = df['b'] + pd.DateOffset(years=1)
df

Out[65]:
            a          b
0  2011-01-31 2011-06-30
1  2011-02-28 2011-07-31
2  2011-03-31 2011-08-31
3  2011-04-30 2011-09-30
4  2011-05-31 2011-10-31
5  2011-06-30 2011-11-30
6  2011-07-31 2011-12-31
7  2011-08-31 2011-01-31
8  2011-09-30 2011-02-28
9  2011-10-31 2011-03-31
10 2011-11-30 2011-04-30
11 2011-12-31 2011-05-31

在一般情况下,您可以使用apply动态应用差异:

In [69]:
df['b'] = df.apply(lambda x: x['b'] + pd.DateOffset(years=x['a'].year-x['b'].year), axis=1)
df

Out[69]:
            a          b
0  2011-01-31 2011-06-30
1  2011-02-28 2011-07-31
2  2011-03-31 2011-08-31
3  2011-04-30 2011-09-30
4  2011-05-31 2011-10-31
5  2011-06-30 2011-11-30
6  2011-07-31 2011-12-31
7  2011-08-31 2011-01-31
8  2011-09-30 2011-02-28
9  2011-10-31 2011-03-31
10 2011-11-30 2011-04-30
11 2011-12-31 2011-05-31