Python: удалить строку заголовков CSV и переписать новую строку заголовков

Question

Все еще учу себя Python, поэтому, пожалуйста, не ненавидь меня, если мой код ужасен ...

Мой код:

ifile = csv.reader(open("TE.csv",'rb'))
shutil.copy("TE.csv","temp")
tempfile = csv.reader(open("temp","rb"))
ofile = csv.writer(open("TE-RESULTS.csv","ab"))

for row in ifile:
    #do some web scraping stuff here

    VC_s = str(cells[1].find(text=True))
    VC_i = str(cells[2].find(text=True))
    VT_s = str(cells[4].find(text=True))

    entry = [VC_s, VC_i, VT_s]

    rowAdd = tempfile.next()
    ofile.writerow(rowAdd + entry)

Проблема: я начинаю с CSV, которыйдолжны добавить 3 столбца до конца.Используя мой код выше, я получаю следующий вывод:

HEADER1     HEADER2    HEADER3   result1   result2   result3
autocheck   C:\check   1.jpg     result1   result2   result3
services    C:\svcs    2.jpg     result1   result2   result3

Мой Желаемый вывод:

HEADER1     HEADER2    HEADER3   HEADER4   HEADER5   HEADER6
autocheck   C:\check   1.jpg     result1   result2   result3
services    C:\svcs    2.jpg     result1   result2   result3

Каков наилучший способисправить мой код, который даст мне желаемый результат?Сначала я хотел удалить строку HEADERS и заменить ее новой строкой HEADERS в файле TE-RESULTS.csv.

Answer 1

Обработка строки заголовка перед циклом for.

rowAdd = tempfile.next()
ofile.writerow(rowAdd + ['HEADER4', 'HEADER5', 'HEADER6'])

for ...