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

Question

嗨，我想了解pd.DataFrame.idxmax效率pd.DataFrame.idxmax以查看是否值得用可能更高效的自定义算法替换（例如，使用二进制搜索）。

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

Answer 1

根据消息来源，作者声明This method is the DataFrame version of ndarray.argmax 。 此方法argmax具有的时间复杂度O(N)如图这里。 然后假设pd.DataFrame.idxmax具有相同的时间复杂度是合理的。

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