基于某些条件熊猫的列中的增量计数器

时间:2020-05-29 09:41:53

标签: python pandas dataframe

col1     col2
 -        -
 -        -
 no       1
 -        -
 no       2
 no       3

我在数据框中有2列。每当在col1中遇到“否”时,都需要如上所述在col2中增加计数器

1 个答案:

答案 0 :(得分:1)

==的{​​{3}}值乘以Series.eq,然后使用累积总和,用Series.where将非no的值替换为-

m = df['col1'].eq('no')
df['col2'] = m.cumsum().where(m, '-')
print (df)
  col1 col2
0    -    -
1    -    -
2   no    1
3    -    -
4   no    2
5   no    3