熊猫系列累积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

上的文档