dask_expr._collection.Series.random_split

dask_expr._collection.Series.random_split

Series.random_split(frac, random_state=None, shuffle=False)

Pseudorandomly split dataframe into different pieces row-wise

Parameters
fraclist

List of floats that should sum to one.

random_stateint or np.random.RandomState

If int or None create a new RandomState with this as the seed. Otherwise draw from the passed RandomState.

shufflebool, default False

If set to True, the dataframe is shuffled (within partition) before the split.

See also

dask.DataFrame.sample

Examples

50/50 split

>>> a, b = df.random_split([0.5, 0.5])  

80/10/10 split, consistent random_state

>>> a, b, c = df.random_split([0.8, 0.1, 0.1], random_state=123)