PHP前端开发

如何使用 Pandas 统计当前行值之前比该值大的数据的个数?

百变鹏仔 5天前 #Python
文章标签 如何使用

如何使用 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))]

代码说明:

最终,你会得到一个新的 'col4' 列,其中包含从当前行 'col3' 值到向上第一个比该值大的值之间的计数。