由网友(★小丑)分享简介：我想知道是否有可能在GroupBy Aggregation with Dask.之后从给定列中获得唯一项的数量。我在文档中没有看到任何类似的内容。它可以在 pandas 数据框上使用，并且真的很有用。我看到了一些与此相关的问题，但我不确定它是否已实施。有人能给我一些提示吗？推荐答案若要展开this comment，可...

我想知道是否有可能在GroupBy Aggregation with Dask.之后从给定列中获得唯一项的数量。我在文档中没有看到任何类似的内容。它可以在 pandas 数据框上使用，并且真的很有用。我看到了一些与此相关的问题，但我不确定它是否已实施。

有人能给我一些提示吗？

推荐答案

若要展开this comment，可以直接在序列组上使用nunique：

import pandas as pd
import dask.dataframe as dd

d = {'col1': [1, 2, 3, 4], 'col2': [5, 6, 7, 8]}
df = pd.DataFrame(data=d)
ddf = dd.from_pandas(df, npartitions=2)
ddf.groupby(['col1']).col2.nunique().to_frame().compute()

有关详细讨论，请参阅https://github.com/dask/dask/issues/6280。