熊猫：创建一列，使另一列中的元素计数等于某个值

Question

我有一列具有从1到200的不同值的列。元素的顺序很重要，但是它们并不是根据其值排序的。我想要一个新列，其中包含的元素数等于200。

示例：

A包含要计数的元素

B包含计数。

A    B
10   0
20   0
50   0
200  1
15   1
68   1
200  2
199  2
15   2
200  3
1    3
200  4

我的方法是这样的：

df['B'] = np.where(df['A'] == 200, 1, 0)
df['B'] = df['B'].cumsum()

这是正确的吗？是否存在更好的方法？