如何使用 pandas 获取比当前行值大的数据的计数
要获取比当前行值大的数据的计数，可以使用 pandas 提供的矩阵计算。这种方法可以避免查找循环，从而将时间复杂度降低到 o(1)。
代码示例如下：
count_larger = lambda x: np.sum(x > x[-1]) df['col4'] = [count_larger(df[:i + 1]['col3'].values) for i in range(len(df))]
代码说明：
- count_larger 函数是一个 lambda 表达式，它计算一个数组中大于其最后一个元素的值的数量。
- df[:i + 1]['col3'].values 获取从第 0 行到第 i 行的 'col3' 列的值，并将其转换为 numpy 数组。
- 对于每一行，count_larger 函数计算比该行 'col3' 值大的值的计数。
- [count_larger(df[:i + 1]['col3'].values) for i in range(len(df))] 使用列表推导为每一行计算计数，并将其作为 'col4' 列添加到 dataframe 中。
最终，你会得到一个新的 'col4' 列，其中包含从当前行 'col3' 值到向上第一个比该值大的值之间的计数。
