DataFrame to Json с использованием первого столбца в качестве ключа и второго в качестве значения

Question

У меня есть фрейм данных, который выглядит так:

k = pd.DataFrame({'A':[1,2,3,4], 'B':['a','b','c','d']})

И я хочу вставить в mongoDB, похожую на это:

dic = {1:'a', 2:'b',3:'c',4:'d'}

Как я могу это сделать?

Я проверил подобные вещи, но они, похоже, не работают на моем df:

конвертирование pandas dataframe в объект json - pandas

Заранее спасибо!

Answer 1

Хотя то, что я пишу, не является ответом на заданный вопрос, тем не менее я даю решение небольшой проблемы, с которой я столкнулся, которую я нашел и нашел.

Проблема: как создать словарь из фрейма данных panda со столбцом в качестве ключа и постоянным значением (1 в моем случае) в качестве, как вы уже догадались, значения.

Решение:

f = pd.Series(data = [1]*df.shape[0],index=df['col_name'])
x = f.to_json(orient='columns')

Вывод:

{"one":1, "two":1, "three": 1}

Зачем мне это делать?Поскольку поиск по словарю высоко оптимизирован (да, я тоже могу использовать set)

PS Новичок в Python, поэтому, пожалуйста, будьте осторожны со мной:).

Answer 2

Используйте Series.to_json и при необходимости измените key значение добавьте rename:

print (k.set_index('A').rename(columns={'B':'index1'}).to_json())
{"index1":{"1":"a","2":"b","3":"c","4":"d"}}

Если необходимо экспортировать в файл:

k.set_index('A').rename(columns={'B':'index1'}).to_json('file.json')