在数据栏中标注相同的日期？

Question

我有一个有日期的数据：

df：

date     value
20210403 1 
20210403 33
20210401 33
20210408 44
20210401 -11 
20210402 333

我想标注相同的日期并得到输出：

date     value  date_label
20210403 1               1
20210403 33              1
20210401 33              2
20210408 44              3
20210401 -11             2 
20210402 333             4

我考虑使用df.date.unique()并添加一列标签，然后将其添加到df中。有什么内置的功能可以做吗？

Answer 1

下面是您可以尝试的解决方案，enumerate关于唯一日期值，并将其返回到原始df。

map_ = {j: i for i, j in enumerate(df['date'].unique(), 1)}

df['date_label'] = df['date'].map(map_)

       date  value  date_label
0  20210403      1           1
1  20210403     33           1
2  20210401     33           2
3  20210408     44           3
4  20210401    -11           2
5  20210402    333           4