Словари и списки в файлах

[Галина_Д]

Доброго времени суток. У меня есть файл, который я перевела в словарь формата ключ-это id, значение-category.
Далее нужно другой файл открыть для чтения и сразу писать еще файл, добавив столбец с категорией из словаря (ищем по id).
Вроде все сделала, но проблема в том, что у всех id на выходе одна и та же категория.
На всякий случай даю оба кода.
Код перевода в словарь:

import json
i = 0
purchases_new = {}
with open('purchase_log.txt') as purchases:
  for line in purchases:
    line = line.strip()
    purchases_dict = json.loads(line)
    key = purchases_dict['user_id']
    value = purchases_dict['category']
    if key not in purchases_new:
      purchases_new[key] = value

Код записи:

i = 0
with open('visit_log.csv', 'r') as visit:
  with open('funnel.csv', 'w') as visit_write:
    for visit_line in visit:
      visit_list = visit_line.strip().split(',') 
      if visit_list[0] in purchases_new.keys():
        visit_list.append(purchases_new[key])
        visit_write.write(str(visit_list)+'\n')
        print(visit_list)
        i += 1
        if i > 5:
          break

[BDA]

Кажется, что нужно так:

visit_list.append(purchases_new[visit_list[0]])

[Галина_Д]

Спасибо огромное, а-то я уже как только не писала... Но такого варианта и в голову не приходило даже.