gpt4 book ai didi

python - 在 Pandas 中按组创建递增的整数系列

转载 作者:太空宇宙 更新时间:2023-11-04 08:11:49 25 4
gpt4 key购买 nike

我正在尝试在 pandas 中按组创建一个整数升序列表:

Group    Date
A 2000-01-01
A 2000-01-12
A 2000-01-15
A 2000-10-01
B 2005-02-05
B 2006-04-10
B 2010-08-20

会变成这样:

Group    Date          obs_num
A 2000-01-01 1
A 2000-01-12 2
A 2000-01-15 3
A 2000-10-01 4
B 2005-02-05 1
B 2006-04-10 2
B 2010-08-20 3

最佳答案

您还可以使用 cumsum 做一些巧妙的事情,例如:

df['obs_num'] = 1
df['obs_num'] = df.groupby('Group')['obs_num'].cumsum()

我的小测试中这个版本为 6 毫秒,而汤姆的解决方案为 14.8 毫秒。

关于python - 在 Pandas 中按组创建递增的整数系列,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/21005925/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com