Pandas dataframe to_dict два столбца как индексы и третий как значение
У меня есть фреймворк pandas, как показано ниже:
User ASIN Rating
A23VKINWRY6J92 1476783284 5
A3HC4SRK7B2AXR 1496177029 5
AE12HJWB5ODOD B00K2GAUC0 4
AL4RYO265J1G 061579615X 3
Я хочу создать словарь, который имеет 2 столбца «Пользователь» и «ASIN» в качестве ключей и третий столбец «Рейтинг» в качестве значения. Что-то вроде ниже:
my_dict[A23VKINWRY6J92][1476783284] = 5
my_dict[A3HC4SRK7B2AXR][1496177029] = 5
my_dict[AE12HJWB5ODOD][B00K2GAUC0] = 4
my_dict[AL4RYO265J1G][061579615X] = 3
Как я могу это сделать?
Может ли один User иметь несколько ASIN?
Вы можете использовать set_index, что-то вроде df.set_index (['User', 'ASIN']). Rating.to_dict (). Здесь ключи будут в виде кортежей, и вы сможете получить к ним доступ, используя my_dict [('A23VKINWRY6J92', '1476783284')]
Ответы 4
Использование вложенного понимания dict:
{u: {a: list(df.Rating[(df.User == u) & (df.ASIN == a)].unique()) for a in df.ASIN[df.User == u].unique()} for u in df.User.unique()}
Обратите внимание, что это отображается в списки, поскольку нет причин, по которым результирующее значение должно быть уникальным.
Хороший ответ ... :)
Ваш вопрос не слишком ясен, но делает ли это то, что вы хотите?
>>> D = df.groupby(['User','ASIN'])['Rating'].apply(list).to_dict()
>>> {key[0]:{key[1]:val} for key, val in D.items()}
{('A23VKINWRY6J92', '1476783284'): [5], ('A3HC4SRK7B2AXR', '1496177029'): [5], ('AE12HJWB5ODOD', 'B00K2GAUC0'): [4], ('AL4RYO265J1G', '061579615X'): [3]}
Итак, если это назначено на my_dict, тогда у вас есть
>>> my_dict['A23VKINWRY6J92']['1476783284']
[5]
так далее
Мне нравится этот ответ. Лично я бы заменил max на apply(list).
@AmiTavory да, но это не сработает, если на ASIN будет несколько User, тогда как ваш будет ...
:-) Тогда мои плохие навыки Pandas окупаются.
Это должно работать, пока у вас есть уникальный идентификатор пользователя.
my_dict = {d['ASIN'] : {d['User'] : d['Rating']} for d in df.to_dict(orient='records')}
В качестве альтернативы вы можете отфильтровать DataFrame, чтобы получить рейтинг
rating = df.loc[(df['User']=='A23VKINWRY6J92') & (df['ASIN']=='1476783284'), 'Rating'][0]
Можно использовать defaultdict
from collections import defaultdict
d = defaultdict(dict)
for _,x in df.iterrows():
d[x['User']][x['ASIN']] = x['Rating']
d=dict(d)
d['A23VKINWRY6J92']['1476783284']
Out[108]: 5
Это сработало, спасибо! Как мне инициализировать словарь до 0, чтобы я не получал KeyError для недопустимых ключей?
Что бы вы хотели, если бы два столбца были одинаковыми?