Как перевести столбец фрейма данных в формат, который нужен tslearn? - PullRequest
1 голос
/ 13 апреля 2019

Я учусь 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]]

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

1 Ответ

2 голосов
/ 13 апреля 2019

Попробуйте:

a.groupby('cla').agg(list)['value'].to_list()

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...