Which pandas methods have engine_kwargs for numba support?

Question

This pandas doc notes that numba can be used "in select pandas methods," and, "Methods that support engine="numba" will also have an engine_kwargs keyword...."

What are these select pandas methods? Is there a list somewhere?

(Note that this is not trivial: The example provided on that page shows the argument being passed to .apply(), but DataFrame.apply() does not support it. Is it DataFrame.rolling? Nope. In that example it is derived from window.rolling....)

Answer 1

Here is an (hopefully) exhaustive list of pandas function that support engine="numba".

Top level pandas functions:

• pandas.eval

Groupby operations (DataFrame and Series):

• pandas.core.groupby.DataFrameGroupBy.mean, sum, median, var, std, min, max
• agg
• transform
• apply

Rolling operations:

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

• apply

Expanding operations

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

Exponential moving window (ewm) operations:

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