我有一个数据帧-mydata-包含500行是这样的:
Id Name Score
R1 sam 76
R1 Sosan 8
.. … …
R4 jack 2
R4 Tom 76
R4 samy 8
R5 Check 9
… …
现在,我想将数据帧转换成字典,以便潮头IDS成为键和唯一身份标识的行成为一个列表。例如,考虑上面的例子中,R1应该是一个关键和R1的内部,我们应该有两个列表,或R4,我们应该有三个列表等我有TREID下面的代码,但它忽略的唯一ID:
mydictest= mydata.set_index('Id').T.to_dict('list').copy()
所以,我的期望输出应该是这样的:
{'R1': [['sam', 78],['Sosan',8]], 'R4': [['Jack', 2],['Tom', 76],
['samy', 8]]}
也许你可以尝试这样的事:
df = pd.DataFrame([['R1', 'sam', 78], ['R2', 'rem', 65], ['R1', 'pem', 56]], columns=['Id', 'Name', 'Score'])
d = df.set_index('Id').to_dict(orient='split')
req_dict = {}
for ind, data in zip(d['index'], d['data']):
if ind in req_dict:
req_dict[ind].append(data)
else:
req_dict[ind] = [data]
print(req_dict)
输出:
{'R1': [['sam', 78], ['pem', 56]], 'R2': [['rem', 65]]}