使用 cython/numba 进行 Pandas groupby 聚合

Question

使用DataFrame.apply，可以使用raw=True，这会导致使用 Numpy 数组调用应用的函数，通过使用带有 Numba 的 JITint 在 Cython 中编写函数来实现速度。

DataFrame.groupby().agg 也可以吗？

Answer 1

很遗憾，目前这是不可能的。

GroupBy.agg 接受单个参数func，这是一个可调用的（您要运行的函数）。任何其他 args 或 kwargs 都将传递给您的 func。

见：pandas docs