Как перевести столбец фрейма данных в формат, который нужен tslearn?

Я изучаю tslearn. Но я действительно смущен форматом входных данных. Допустим, есть фрейм данных:

a=pd.DataFrame({'cla':['a','a','a','a','b','b','b'],
                'value':[1,2,3,4,2,4,1]})

Я хочу, чтобы a и b были отдельными временными рядами соответственно. Итак, это выглядит так:

a: 1,2,3,4
b: 2,4,1

Согласно tslearn, пример ввода:

X = to_time_series_dataset([[1, 2, 3, 4], [1, 2, 3], [2, 5, 6, 7, 8, 9]])

Итак, как я могу передать данные:

[[1,2,3,4],[2,4,1]]

Большое спасибо.