Блог

У меня есть код, который берет JSON из URL-адреса и сохраняет его в CSV-файл.
Второй код берет только одну строку из CSV-файла и ставит разделитель»,».

 --Takes the JSON from URL and converts to CSV

res = conn.getresponse()
data = res.read()
data_f = data.decode('utf-8')
data_j = json.loads(data_f)['results']s']

with open("../lookups/customer_id.csv", 'w', encoding='utf-8', newline='') as f:
    wr = csv.DictWriter(f, fieldnames = data_j[0].keys()) 
    wr.writeheader() 
    wr.writerows(data_j) 

 

Результат, который я получаю, таков:

имя, идентификатор
aaa, 111
bbb, 222
ccc, 333
ddd, 444
…

Второй скрипт берет файл (customer_id.csv) и преобразует его в этот формат:

111,222,333,444

 import csv, re

with open("../lookups/customer_id.csv", newline='', encoding='utf-8') as f:
    reader = csv.reader(f)
    next(reader, None)
    c=""
    for row in reader:
        c = c "," row[1]
get_customer = re.sub('^,', '', c)

file1 = open("../lookups/customer_no_list", 'w')
file1.write(get_customer)
file1.close()
 

Я хотел бы объединить его в один файл. Мне не нужен файл customer_id.csv, только файл customer_no_list.
Поскольку мои навыки работы на Python не так хороши, что может быть лучшим решением?

Спасибо