我有一个数据帧,如下所示:
df = pd.DataFrame({'State': ['Texas', 'Texas', 'Florida', 'Florida'],
'a': [4, 5, 1, 3], 'b': [6, 10, 3, 11]})
df['ts'] = datetime.utcnow()
表格如下所示
State a b ts
0 Texas 4 6 2022-09-06 15:33:31
1 Texas 5 10 2022-09-06 15:33:31
2 Florida 1 3 2022-09-06 15:33:31
3 Florida 3 11 2022-09-06 15:33:31
我想要实现的是,对于每个组‘ts’应该是唯一的,所以我想用+1秒递增它的所有其他值 因此,输出数据帧将如下所示:
State a b ts
0 Texas 4 6 2022-09-06 15:33:31
1 Texas 5 10 2022-09-06 15:33:32
2 Florida 1 3 2022-09-06 15:33:31
3 Florida 3 11 2022-09-06 15:33:32
使用GROUPBY和Transform,可以获得系列,但不能进一步:
df['ts'] = df['ts'].groupby(df['State']).transform(lambda x: increment_ms(x))
我如何才能实现上述输出?