pandas - 哪些 pandas 方法具有 engine_kwargs 来支持 numba？

Question

This pandas doc注意到 numba 可以“在选择的 pandas 方法中使用”，并且“支持 engine="numba" 的方法也将使用有一个 engine_kwargs 关键字...."

这些选择的 pandas 方法是什么？某处有列表吗？

(请注意，这并不简单:该页面上提供的示例显示了传递给 .apply() 的参数，但 DataFrame.apply() 没有支持它。是 DataFrame.rolling 吗？不。在该示例中，它派生自 window.rolling....)

Answer 1

这里是支持 engine="numba" 的 pandas 函数(希望如此)详尽的列表。

顶级 pandas 函数:

• pandas.eval

Groupby 操作(DataFrame 和 Series):

• pandas.core.groupby.DataFrameGroupBy.mean , sum , 中位数, var , std , min , max
• agg
• transform
• 申请

滚动操作:

• pandas.core.window.rolling.Rolling.mean , sum , median , var , std , min , max

• apply

扩大业务

• pandas.core.window.expanding.Expanding.mean , sum , median , var , std , min , max
• apply

指数移动窗口(ewm)运算:

• pandas.core.window.ewm.ExponentialMovingWindow.mean , sum , var , std , min , max
• apply