Welcome to OStack Knowledge Sharing Community for programmer and developer-Open, Learning and Share
Welcome To Ask or Share your Answers For Others

Categories

0 votes
1.2k views
in Technique[技术] by (71.8m points)

python - Pandas groupby cumulative count

How do I do a cumcount on two groupings, id and name? For example I have:

id  name
1   name1
1   name1
1   name2
2   name3
2   name3
2   name3

And I would like it to look like:

id  name   cumcount
1   name1  1 
1   name1  2
1   name2  1
2   name3  1
2   name3  2
2   name3  3

Basically a running tally for each name in an id.

See Question&Answers more detail:os

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome To Ask or Share your Answers For Others

1 Answer

0 votes
by (71.8m points)

You can use groupby.cumcount:

df['cumcount'] = df.groupby(['id', 'name']).cumcount() + 1

df
#  id   name    cumcount
#0  1   name1   1
#1  1   name1   2
#2  1   name2   1
#3  2   name3   1
#4  2   name3   2
#5  2   name3   3

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome to OStack Knowledge Sharing Community for programmer and developer-Open, Learning and Share
Click Here to Ask a Question

...