Как мне выполнить разделение поездов и тестов в кадре данных Polars? ⇐ Python

[Anonymous]

Я пытаюсь найти простой способ случайного разделения кадра данных Polars при обучении и тестировании. Вот как я сейчас это делаю

Код: Выделить всё

train, test = df
.with_columns(pl.lit(np.random.rand(df0.height)>0.8).alias('split'))
.partition_by('split')

однако в моих фреймах данных остается дополнительный разделенный столбец, который мне нужно удалить.

Подробнее здесь: https://stackoverflow.com/questions/764 ... -dataframe