如何在考拉的 groupby 对象中获取组数?

How to get number of groups in a groupby object in koalas?

如何获取 koalas 中的 groupby 对象中的组数?

在pandas中我们可以使用ngroups,但是这个方法在考拉中还没有实现。

假设 groupby 对象被调用 dfgroup

有什么想法吗?

您需要在 GroupBy object(类型:databricks.koalas.groupby.DataFrameGroupBy)上强制执行某些函数,并从中获取长度。例如,dfgroup.size() 将 return databricks.koalas.series.Series,您可以在其上调用 len(示例改编自文档):

>>> grouped = df.groupby(['Animal'])
>>> type(grouped.size())
<class 'databricks.koalas.series.Series'>
>>> len(grouped.size())
2