pandas `idxmax` 方法使用的时间复杂度/算法

Question

您好，我想了解如何有效地pd.DataFrame.idxmax查看是否值得用可能更有效的自定义算法替换（例如使用二分搜索）。我想了解这种方法背后的算法，或者至少了解它的复杂性，但到目前为止我还没有运气。任何帮助将不胜感激，谢谢。

Answer 1

This method is the DataFrame version of ndarray.argmax。此方法argmax的时间复杂度O(N)如下所示。那么可以合理地假设pd.DataFrame.idxmax具有相同的时间复杂度。

如果您想实现自己的搜索算法，请记住，您建议的二分搜索需要一个排序的项目数组，而列的情况可能并非如此DataFrame。