创建根据其他列值增加计数的数据框列

时间:2020-11-02 18:59:10

标签: python pandas dataframe

我需要在数据框中添加一列,每当另一列中的值超过限制时,就会添加一个数字。下面的示例:

原始DF:

ColA   ColB
 4      1.4
 10     0.5
 1      2.3
 3      12.2
 8.8    8.5
 2      5.2
 0.6    0.33
 9      3
 4      144
 33     8

所需DF:ColB中的值超过10时,ColC计数=计数+1

ColA   ColB     ColC
 4      1.4       1
 10     0.5       1
 1      2.3       1
 3      12.2      2
 8.8    8.5       2
 2      5.2       2
 0.6    0.33      2
 9      3         2
 4      144       3
 33     8         3

感谢您的帮助!

1 个答案:

答案 0 :(得分:1)

df['ColC'] = (df['ColB'] > 10).cumsum() + 1
print(df)

打印:

   ColA    ColB  ColC
0   4.0    1.40     1
1  10.0    0.50     1
2   1.0    2.30     1
3   3.0   12.20     2
4   8.8    8.50     2
5   2.0    5.20     2
6   0.6    0.33     2
7   9.0    3.00     2
8   4.0  144.00     3
9  33.0    8.00     3