diff --git a/eds_scikit/utils/sort_first_koalas.py b/eds_scikit/utils/sort_first_koalas.py index 26414fa9..52508f45 100644 --- a/eds_scikit/utils/sort_first_koalas.py +++ b/eds_scikit/utils/sort_first_koalas.py @@ -6,9 +6,9 @@ def sort_values_first_koalas( dataframe: DataFrame, by_cols: List[str], cols: List[str], ascending: bool = True ) -> DataFrame: - """Replacement for dataframe.sort_values(cols).groupby(by_cols).first() + """Use this function to obtain in koalas the same ouput as dataframe.sort_values(cols).groupby(by_cols).first() in pandas. - To get a deterministic first, provide an id column of your dataframe as the last element of cols. + If you want the output to be deterministic, provide an id column of your dataframe as the last element of variable cols. Parameters ----------