pandas：如何生成新列以获取另一列的最后日期时间值

Question

我有一个如下的数据框：

obj_id   data_date   value
4        2010-11-01  59500    
2        2011-10-01  35200 
4        2012-07-31  24860   
1        2013-07-28  15860
2        2014-10-15  200200

如何生成一个新列来记录对象的最后时间值？如果一个对象之前没有出现过，它应该将默认值设置为0。输出应该是这样的：

obj_id   data_date   value    last_value
4        2010-11-01  59500    0
2        2011-10-01  35200    0
4        2012-07-31  24860    59500
1        2013-07-28  15860    0
2        2014-10-15  200200   35200

Answer 1

你需要groupby().shift()

df['last_value'] = df.groupby('obj_id')['value'].shift(fill_value=0)

输出：

   obj_id   data_date   value  last_value
0       4  2010-11-01   59500           0
1       2  2011-10-01   35200           0
2       4  2012-07-31   24860       59500
3       1  2013-07-28   15860           0
4       2  2014-10-15  200200       35200