pandas df为稀疏矩阵的嵌套字典

时间:2018-02-25 19:12:34

标签: python pandas dictionary group-by sparse-matrix

我正在寻求将我的数据帧转换为嵌套字典,以便稍后将数据配置为类似艺术家的稀疏矩阵。我下面的尝试几乎完成,但我不确定如何正确地将嵌套列表更改为字典。

以下是示例数据:

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}}

0 个答案:

没有答案