Creating a new column from another column + unique numeric index in pandas dataframe

Question

I have a dataframe in which several rows of a column have the same value:

   unique_code      0
0   p01_PAR_1  zertara
1   p01_PAR_1    atera
2   p01_PAR_1       da
3   p01_MOT_1       ez
4   p01_MOT_1    dakit

I would like to redo that column or create a new one with one unique value, by adding a numeric index after the value, so it would result on something like this:

   unique_code       0
0   p01_PAR_1_1  zertara
1   p01_PAR_1_2    atera
2   p01_PAR_1_3       da
3   p01_MOT_1_1       ez
4   p01_MOT_1_2    dakit

This cannot be done by adding the row index to each row, since they have different and unrelated values.

Erfan · Accepted Answer · 2020-04-21 15:22:46Z

2

Use GroupBy.cumcount then add it as a string:

df['unique_code'] = (
    df['unique_code'] + 
    '_' + 
    df.groupby('unique_code').cumcount().add(1).astype(str)
)

   unique_code        0
0  p01_PAR_1_1  zertara
1  p01_PAR_1_2    atera
2  p01_PAR_1_3       da
3  p01_MOT_1_1       ez
4  p01_MOT_1_2    dakit

answered Apr 21, 2020 at 15:22

Erfan

43.4k10 gold badges76 silver badges86 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Creating a new column from another column + unique numeric index in pandas dataframe

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related