python - 将数据框转换为子列表或字典

Question

我有表格格式的数据(行和列)，我将其读入数据框(Data1):

     Name         D    Score
0   Angelica     D1     3.5
1   Angelica     D2     2.0
2   Bill         D1     2.0
3   Chan         D3     1.0 
......

我可以使用以下方法将其转换为列表:

Data2 = Data1.values.tolist()

并得到以下输出:

[
['Angelica', 'D1', 3.5], ['Angelica', 'D2', 2.0],
['Bill', 'D1', 2.0], ['Bill', 'D2', 3.5],
['Chan', 'D8', 1.0], ['Chan', 'D3', 3.0], ['Chan', 'D4', 5.0],
['Dan', 'D4', 3.0], ['Dan', 'D5', 4.5], ['Dan', 'D6', 4.0]
]

我想要的是，输出是这样的:

{
'Angelica': {'D1': 3.5, 'D2': 2.0} ,
'Bill': {'D1': 2.0, 'D2': 3.5} 
'Chan': {'D8': 1.0, 'D3': 3.0, 'D4': 5.0 } 
'Dan': {'D4': 3.0, 'D5': 4.5, 'D6': 4.0} 
}

如何在 Python 中实现这一点？

Answer 1

您可以在分组 df 之后使用字典理解 Name 列:

>>> df = pd.DataFrame([{'Name': 'Angela', 'Score': 3.5, 'D': 'D1'}, {'Name': 'Angela', 'Score': 2.0, 'D': 'D2'}, {'Name': 'Bill', 'Score': 2.0, 'D': 'D1'}, {'Name': 'Chan', 'Score': 1.0, 'D': 'D3'}])
>>> df
    D    Name  Score
0  D1  Angela    3.5
1  D2  Angela    2.0
2  D1    Bill    2.0
3  D3    Chan    1.0
>>> data2 = {name: {df.ix[v].D: df.ix[v].Score for v in val} for name, val in df.groupby('Name').groups.items()}
>>> data2
{'Chan': {'D3': 1.0}, 'Angela': {'D1': 3.5, 'D2': 2.0}, 'Bill': {'D1': 2.0}}