类型提示 Pandas DataFrameGroupBy

Question

我应该如何在 Python 一个 pandas DataFrameGroupBy 对象中键入提示？ 对于正常的 pandas 数据帧，我应该只使用 pd.DataFrame 吗？ 我没有找到任何其他解决方案 atm

Answer 1

DataFrameGroupBy 本身就是一个合适的类型。因此，如果您正在编写一个必须专门采用 DataFrameGroupBy 实例的函数：

from pandas.core.groupby import DataFrameGroupBy

def my_function(dfgb: DataFrameGroupBy) -> None:
    """Do something with dfgb."""

如果您正在寻找更通用的多态类型，有几种可能性：

• pandas.core.groupby.GroupBy 因为 DataFrameGroupBy 继承自 GroupBy[DataFrame].
• 如果你也想接受 Series 个实例，你可以联合 DataFrameGroupBy 和 SeriesGroupBy 或者你可以使用 GroupBy[FrameOrSeries] （如果你打算总是匹配在您的 return 值中输入类型）或 GroupBy[FrameOrSeriesUnion] 如果您的输出类型不反映输入类型。所有这些类型都在 pandas.core.groupby.generic.
中
• 您可以根据自己的喜好以多种不同方式组合上述泛型（和其他泛型）。