用于熊猫`idxmax`方法的时间复杂度/算法

Question

您好，我想了解pd.DataFrame.idxmax 的效率有多高，看看是否值得用可能更有效的自定义算法替换（例如，使用二分搜索）。

我想了解这种方法背后的算法，或者至少了解它的复杂性，但是到目前为止我还没有运气。任何帮助将不胜感激，谢谢。

Answer 1

根据source，作者声明This method is the DataFrame version of ndarray.argmax。此方法argmax 的时间复杂度为O(N)，如here 所示。然后可以合理地假设pd.DataFrame.idxmax 具有相同的时间复杂度。

如果您想实现自己的搜索算法，请记住，您建议的二分搜索需要排序的项目数组，DataFrame 列可能不是这种情况。