python - 将行与之前的所有行进行比较

Question

我有一个这样的数据框，

df = pd.DataFrame({'numb': [2,4,6,2,4,9]})
print(df)
   numb
0     2
1     4
2     6
3     2
4     4
5     9

我想计算 numb 小于前一个 numb 的行数，比如

   numb  lesser_count
0     2             0
1     4             0
2     6             0
3     2             2
4     4             1
5     9             0

我遇到了类似的 example但不确定如何计数。

Answer 1

让我们试试 numpy 广播:

a = df['numb'].to_numpy()

df['lesser_count'] = np.triu(a<a[:,None]).sum(0)

输出:

   numb  lesser_count
0     2             0
1     4             0
2     6             0
3     2             2
4     4             1
5     9             0