중간 복사본을 만들지 않고 csvfile에 행 추가

Question

위치를 편집하여 CSV 파일에 행을 추가하려면 어떻게해야합니까? 나는 원래, (의사)을 임시 파일에 기록한 다음, 교체의 패턴을 피하려는 :

add_records_to_csv(newdata, infile, tmpfile) 
delete(infile) 
rename(tmpfile, infile) 

는 여기에 실제 기능입니다. 기존의이 기록으로 새로운 행이 동일한 구조를 가지고 있다면

이

def add_records_to_csv(dic, csvfile): 
    """ Append a dictionary to a CSV file. 
     Adapted from http://pymotw.com/2/csv/ 
    """ 
    f_old = open(csvfile, 'rb')       # <-- 
    csv_old = csv.DictReader(f_old)      # <-- 

    fpath, fname = os.path.split(csvfile)    # <-- 
    csvfile_new = os.path.join(fpath, 'new_' + fname) # <-- 
    print(csvfile_new)         # <-- 
    f = open(csvfile_new, 'wb')       # <-- 

    try: 
     fieldnames = sorted(set(dic.keys() + csv_old.fieldnames)) 
     writer = csv.DictWriter(f, fieldnames=fieldnames) 
     headers = dict((n,n) for n in fieldnames) 
     writer.writerow(headers) 
     for row in csv_old: 
      writer.writerow(row) 
     writer.writerow(dic) 
    finally: 
     f_old.close() 
     f.close() 
    return csvfile_new 

Answer 1

은 다음이 작동합니다 : - 라인 "# <"나는 제거 할 및/또는 더 간단 무언가로 응축 무엇

import csv 

def append_record_to_csv(dic, csvfile): 
    with open(csvfile, 'rb') as f: 
     # discover order of field names in header row 
     fieldnames = next(csv.reader(f)) 
    with open(csvfile, 'ab') as f: 
     # assumes that dic contains only fieldnames in csv file 
     dwriter = csv.DictWriter(f, fieldnames=fieldnames) 
     dwriter.writerow(dic) 

반면에 새로운 행을 기존 행과 다른 구조로 작성하면 csv 파일의 형식이 잘못되었을 수 있습니다. csv 파일에 새 열을 추가하려면 모든 행을 편집해야합니다. 이 접근 방식의 성능은 매우 좋지 않으며 대형 CSV 파일에서 눈에 띄게됩니다.

Answer 2

이것은 일반적으로 가능하지 않습니다. 여기 당신의 코드에서, 이유입니다 : 나에게

fieldnames = sorted(set(dic.keys() + csv_old.fieldnames)) 

,이 적어도 어떤 경우에는 새 행이 이전 행에없는 열이 포함되어 있다고 말한다. 이와 같이 행을 추가하면 끝에 새 행을 추가하는 것 외에도 파일의 헤더 (첫 번째 행)를 업데이트해야합니다. 열 이름을 영문자 순서로 지정해야하는 경우 열의 순서를 유지하려면 다른 모든 행의 필드를 다시 정렬해야 할 수 있습니다.

파일의 첫 번째 줄을 편집해야 할 수도 있으므로 마지막 줄에 새 줄을 추가하고 중간에 모든 줄을 편집하는 것 외에도이 작업을 내부에서 수행하는 합리적인 방법이 없습니다 .

내 제안은 프로그램에 헤더를 편집 할 필요가 없으며 단순히 새로운 행을 추가 할 수 있도록하기 위해 포함해야 할 모든 필드/열을 미리 확인하는 것이 좋습니다.