pandasnumba

Which pandas methods have engine_kwargs for numba support?


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....)


Solution

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

    Top level pandas functions:

    Groupby operations (DataFrame and Series):

    Rolling operations:

    Expanding operations

    Exponential moving window (ewm) operations: