Блог

У меня есть csv-файл со следующей структурой:

 1 start,end,ID
2 int1,int2,string1
3 int3,int4,string2
4 int5,int6,string3
5 int7,int8,string4
  

Моя цель — создать новый csv, который записывает уникальную строку для, во-первых, конечного значения и, во-вторых, начального значения в качестве конечного значения.

 1 start,end,ID
2 int1,int2,string1
3 ,int1,string1
4 int3,int4,string2
5 ,int3,string2

  

Я попытался это, записав входной csv-файл в список и выполнив итерацию по этому списку. Для каждой строки в выходной список добавляются две новые строки. После добавления второй строки, соответственно, конечное значение было установлено на начальное значение входного списка. Ниже приведен код, который я использовал:

 import csv

with open(r"input path") as csv_sbw, open("output path","wb") as csv_new:
    csv_in = csv.reader(csv_sbw)
    csv_out = csv.writer(csv_new)
    fields_out = [[]] #list for the output csv
    fields = list(csv_in) #list for the input csv
    fields_out[0] = fields[0] #headline is taken from the input
    fields[0].append("m_value")
    for row in fields[1:]:
        row.append(1)
        if row[2].isdigit() == False and len(row[2]) == 16 and row[2][0] != 0 and row[0] != '' and row[0] != '0' and row[1] != '0': #invalid rows are skipped
            fields_out.append(row) #first row is appended
            fields_out.append(row) #second row is appended
            fields_out[-1][1] = row[0] #the start value of the last appended row is set as an end value
            fields_out[-1][0] = '' #start field of last appended row is deleted
            fields_out[-1][3] = 0
    csv_out.writerows(fields_out) #output csv is written
  

Вместо создания csv в соответствии с примером, описанным выше, я получаю следующий результат:

 1 start,end,ID,m_value
2 1032,1032,'A',0
3 1032,1032,'A',0
4 613,613,'B',0
5 613,613,'B',0
  

Итак, при изменении fields[-1] код, похоже, перезаписывает также предпоследнюю добавленную строку. Насколько я понимаю, добавление двух значений подряд в список создает два новых элемента списка, из которых по моему запросу будет возвращено только последнее добавленное значение list[-1] .
Как мне предотвратить перезапись кодом обеих добавленных строк и вместо этого позволить ему перезаписать только последнюю добавленную строку?

Чтобы подтвердить вашу цель, приведенный ниже фрагмент создает новый csv, который

1. записывает строку для исходной строки и,
2. записывает строку с начальным значением и строкой.

Если эта оценка верна, я обычно открываю по одному файлу за раз, чтобы минимизировать свою когнитивную нагрузку.

 with open('input.csv','r') as file_handle:
    file_content = file_handle.read().split('n')
with open('output.csv','r') as file_handle:
    for index,line in enumerate(file_content):
        if index==0:
            print(line)
            file_handle.write(line)
        else:
            line_as_list = line.split(',')
            print(line_as_list)
            file_handle.write(line)
            print(line_as_list[0], line_as_list[-1])
            file_handle.write(str(line_as_list[0]) "," str(line_as_list[-1]))