通过具有多个列表转换数据帧分成字典?

问题描述 投票:1回答:1

我有一个数据帧-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]]}
python dataframe dictionary converters
1个回答
1
投票

也许你可以尝试这样的事:

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]]}
© www.soinside.com 2019 - 2024. All rights reserved.