Преобразование фрейма данных в словарь, имея несколько списков? - PullRequest
0 голосов
/ 04 февраля 2019

У меня есть датафрейм -mydata-, состоящий из 500 строк, примерно так:

Id  Name    Score
R1  sam     76
R1  Sosan    8
..  …   … 
R4   jack   2
R4  Tom     76
R4   samy    8
R5  Check    9 
…    …                 

Теперь я хочу преобразовать фрейм данных в словарь, чтобы уникальные идентификаторы стали ключами и строкамиуникальные идентификаторы становятся списком.Например, учитывая приведенный выше пример, R1 должен быть ключом, а внутри R1 у нас должно быть два списка, или для R4 у нас должно быть три списка и т. Д. Я обработал приведенный ниже код, но он игнорирует уникальные идентификаторы:

mydictest= mydata.set_index('Id').T.to_dict('list').copy()

Итак, мой желаемый результат должен быть таким:

 {'R1': [['sam', 78],['Sosan',8]], 'R4': [['Jack', 2],['Tom', 76],    
 ['samy', 8]]}

1 Ответ

0 голосов
/ 04 февраля 2019

Может быть, вы можете попробовать что-то вроде этого:

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