hipdf.core.groupby.DataFrameGroupBy.indices

19 min read time

Applies to Linux

property DataFrameGroupBy.indices: dict[ScalarLike, cupy.ndarray]

Dict {group name -> group indices}.

Examples

>>> import cudf
>>> data = [[10, 20, 30], [10, 30, 40], [40, 50, 30]]
>>> df = cudf.DataFrame(data, columns=["a", "b", "c"])
>>> df
    a   b   c
0  10  20  30
1  10  30  40
2  40  50  30
>>> df.groupby(by=["a"]).indices
{10: array([0, 1]), 40: array([2])}