Python的CSV作家 - 寫在新的CSV文件列高達CSV文件

Question

領域的最大數目我有我的文件夾中的200個CSV文件。 我想要做的是讀取每個文件的第一行並寫入新的csv。 和頂部，我想寫[文件，字段1，字段2，...字段N] n是場的最大數量。

import csv 
import glob 
list=[] 
hel=[] 
files=glob.glob('C:/dataset/*.csv') 
with open('test.csv', 'w',newline='') as testfile: 
    csv_writer = csv.writer(testfile) 
    for file in files: 
     with open(file, 'r') as infile: 
      file=file[file.rfind('\\')+1:] 
      file=file.strip('.csv') 
      reader = csv.reader(infile) 
      headers = next(reader) 
      hel.append((len(headers))) 
      max(hel) 
      lst = [file] + headers 
      csv_writer.writerow(lst) 

後來才知​​道的200個文件字段最大數是255 因此，對新的CSV文件的頂部，我想寫file, field1, field2 ... field 255. 我怎樣才能做到這一點？

---

import csv 
import glob 
list=[] 
hel=[] 
files=glob.glob('C:/dataset/*.csv') 
with open('test.csv', 'w',newline='') as testfile: 
    csv_writer = csv.writer(testfile) 
    for file in files: 
     with open(file, 'r') as infile: 
      file=file[file.rfind('\\')+1:] 
      file=file.strip('.csv') 
      reader = csv.reader(infile) 
      headers = next(reader) 
      hel.append((len(headers))) 
      b=['field{}'.format(i) for i in range(1,max(hel)+1)] 
      lst = [file] + headers 
      csv_writer.writerow(lst) 

現在b是列表看起來像這樣[「字段1」，「字段2」 ...「field255」] 我需要插入之前「字段1」文件「，並寫上該行新的csv文件的頂部。在csv_writer.writerow(lst)之後編寫代碼可以讓我隔行掃描'field1','field2'.. csv文件。我該如何解決這個問題

Answer 1

您首先需要讀取所有輸入文件以確定最大字段數爲255.然後，您需要構建一個字段名稱列表以寫入輸出文件（只需一次，而不是在一個循環中）：

['field{}'.format(i) for i in range(1, 256)] 

您可以傳遞一個列表到csv模塊來寫。

Answer 2

讀取每個文件中的字段數和第一行寫入文件前。

import glob 
from itertools import chain 
import os 
from os.path import splitext, basename 

def first_line(filepath): 
    with open(filepath) as f: 
     return next(f) 


def write_test_file(dest_file_path, source_path_name): 
    source_paths = glob.glob(source_path_name) 
    first_lines = list(map(first_line, source_paths)) 

    max_count = max(l.count(",") for l in first_lines) 
    field_names = map("field{}".format, range(1, max_count + 2)) 
    header = ",".join(chain(["file"], field_names)) + os.linesep 

    file_names = (splitext(basename(p))[0] for p in source_paths) 
    content = chain([header], map(",".join, zip(file_names, first_lines))) 

    with open(dest_file_path, 'w') as testfile: 
     testfile.write("".join(content)) 


write_test_file('test.csv', 'C:/dataset/*.csv')