преобразовать фрейм данных в список списка в python - PullRequest
0 голосов
/ 10 апреля 2020

У меня ниже фрейма данных

POC_Number  Resource_Number day_to_complete_task
0   0   12
0   1   12
0   5   6
1   3   8
1   0   7
1   5   5
2   3   9
2   2   11
2   4   10

и я хочу выводить список списков ниже:

[[(0, 12), (1, 12), (5, 6)], [(3, 8), (0, 7), (5, 5)], [(3, 9),(2, 11), (4, 10)]]

Ответы [ 2 ]

0 голосов
/ 10 апреля 2020

Попробуйте это


result = df.groupby('POC_Number')[['Resource_Number', 'day_to_complete_task']].apply(lambda x: x.values.tolist())
result = [list(map(tuple, i)) for i in list(result)]
print(result)

Вывод:

[[(0, 12), (1, 12), (5, 6)], [(3, 8), (0, 7), (5, 5)], [(3, 9), (2, 11), (4, 10)]]
0 голосов
/ 10 апреля 2020

вы можете использовать:

df.groupby('POC_Number').apply(lambda x: [tuple(i[1:]) for i in x.values]).to_list()

или:

df.groupby('POC_Number')['Resource_Number', 'day_to_complete_task'].apply(
    lambda x: [tuple(i) for i in  x.values]).to_list()

выход:

[[(0, 12), (1, 12), (5, 6)],
 [(3, 8), (0, 7), (5, 5)],
 [(3, 9), (2, 11), (4, 10)]]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...