熊猫系列累积argmax

Question

什么是获得另一系列的一系列累加argmax的有效方法？

我想的是

import pandas as pd

def cumargmax(series):
    return pd.expanding_apply(series, lambda x: x.argmax())

这行得通，但我想知道是否有更合适的方法。

Answer 1

您不需要cumargmax函数。 相反，您可以一行完成此操作，

pd.expanding_apply(series, lambda x: x.argmax())

pandas 0.18弃用expanding_apply（）模块级别的功能。 它们将替换为以下方法调用，

series.expanding().apply(lambda x: x.argmax())

在http://pandas.pydata.org/pandas-docs/stable/computation.html上查看文档。