Получить справку по коду Python, который получает данные из одного файла CSV и вставляет их в другой файл CSV

Question

Я делаю миграцию данных. Старые данные приложения экспортируются как один файл CSV. Мы не можем импортировать этот CSV-файл напрямую в новое приложение. Необходимо создать новый шаблон CSV, соответствующий новому приложению, и импортировать некоторые данные в этот новый шаблон CSV. Я хотел бы запросить код, который облегчит это требование.

Answer 1

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

Самое простое - прочитать его построчно и написать:

import csv
with open("Old.csv", 'r') as readfp, open("new.csv", 'w') as writefp:
    csvReader = csv.reader(readfp)
    csvWriter = csv.writer(writefp, delimiter=',')
    for line in csvReader:
        #line is a list of strings so you can reorder it as you wish. I'll skip the third column as an example. 
        csvWriter.writerow(line[:2]+line[3:])

Если у вас установлены панды, это еще проще

import pandas as pd
df = pd.read_csv("Old.csv")
df.drop(labels=["name_of_bad_col1", "name_of_bad_col2"], sep=',')
df.to_csv("new.csv)

Если вы едете по маршруту панд, обязательно ознакомьтесь с документацией ( read_csv , to_csv )