我正在寻求将我的数据帧转换为嵌套字典,以便稍后将数据配置为类似艺术家的稀疏矩阵。我下面的尝试几乎完成,但我不确定如何正确地将嵌套列表更改为字典。
以下是示例数据:
user artist plays
00001411dc427966b17297bf4d69e7e193135d89 animal collective 1373
00001411dc427966b17297bf4d69e7e193135d89 sting 1323
fffe8c7f952d9b960a56ed4dcb40a415d924b224 none more black 117
fffe8c7f952d9b960a56ed4dcb40a415d924b224 queen 117
这是我的代码:
user_artist_dict = user_artist_plays.groupby('users').artist.apply(list).to_dict()
我设想结果为{‘user1’: {‘artist’: plays, 'artist2': plays, 'artist3': plays}, 'user2': {‘artist’: plays, 'artist2': plays, 'artist3': plays}}