Skip to content

Commit c06130b

Browse files
authored
[BACKPORT] Implements DataFrameGroupBy.sample (#1994) (#1995)
1 parent 820cfd7 commit c06130b

File tree

11 files changed

+640
-13
lines changed

11 files changed

+640
-13
lines changed

docs/source/reference/dataframe/groupby.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@ application to columns of a specific data type.
7171
DataFrameGroupBy.cummin
7272
DataFrameGroupBy.cumprod
7373
DataFrameGroupBy.cumsum
74+
DataFrameGroupBy.sample
7475

7576
The following methods are available only for ``SeriesGroupBy`` objects.
7677

mars/dataframe/groupby/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ def _install():
2525
from .getitem import df_groupby_getitem
2626
from .head import head
2727
from .transform import groupby_transform
28+
from .sample import groupby_sample
2829

2930
for cls in DATAFRAME_TYPE:
3031
setattr(cls, 'groupby', groupby)
@@ -63,6 +64,8 @@ def _install():
6364

6465
setattr(cls, 'head', head)
6566

67+
setattr(cls, 'sample', groupby_sample)
68+
6669
for cls in DATAFRAME_GROUPBY_TYPE:
6770
setattr(cls, '__getitem__', df_groupby_getitem)
6871

0 commit comments

Comments
 (0)