Increment a column based on the condition of another column in python

Question

I have a data frame and I want to create a new column name "new" based on a condition on a different column "col". Create the new column name "new" and count it whenever it finds any value in "col".

  index      col        
    1     2.11.67       
    2       NaN        
    3       NaN        
    4     5.10.56      
    5       NaN        
    6     2.10.67      
    7       NaN        
    8     2.09.67

should result in:-

 index      col        new
    1     2.11.67       1
    2       NaN        NaN
    3       NaN        NaN
    4     5.10.56       2
    5       NaN        NaN
    6     2.10.67       3
    7       NaN        NaN
    8     2.09.67       4

Hi @Barmar I used below code but it throw error df.loc[df["col"].notna(),'new']+=1 — MALAM
– MALAM, Commented Sep 23, 2021 at 18:33

mozway · Accepted Answer · 2021-09-23 18:40:57Z

2

You can use a combination of notna, cumsum, and where:

mask = df['col'].notna()
df['new'] = mask.cumsum().where(mask)

output:

           col  new
index              
1      2.11.67  1.0
2          NaN  NaN
3          NaN  NaN
4      5.10.56  2.0
5          NaN  NaN
6      2.10.67  3.0
7          NaN  NaN
8      2.09.67  4.0

answered Sep 23, 2021 at 18:40

mozway

267k13 gold badges56 silver badges106 bronze badges

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

Collectives™ on Stack Overflow

Increment a column based on the condition of another column in python

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