类型提示 Pandas DataFrameGroupBy

Question

我应该如何在 Python 中输入提示 pandas DataFrameGroupBy 对象？ 我应该像普通熊猫数据框一样使用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]（如果您打算始终匹配返回值中的输入类型）或@987654330 @ 如果您的输出类型不反映输入类型。所有这些类型都在pandas.core.groupby.generic 中。
• 您可以根据自己的喜好以多种不同方式组合上述泛型（和其他泛型）。