添加一個新列到.csv與文件名

Question

填充我是比較新的蟒蛇，我試圖將新列添加到所有的.csv文件的目錄，並與家長CSV文件的文件名進行填充。

我的代碼（來自論壇的其他部分）允許我在單個文件（listing1.csv）中添加一個新列並將其稱爲'FileName'，然後使用文件內的其他變量（行0 ，1,2等）。

我想提出兩個更改爲：

1. 從另一列添加的實際文件名的文件（listings1），以文件名，而不是數據
2. 運行本作中的所有CSV文件目錄（listings1，listings2，listings3 ...），只是更新源文件，而不是創建新的（輸出1等）

任何意見讚賞

import csv 

with open('listings1.csv','r') as csvinput: 
with open('output1.csv', 'w') as csvoutput: 
    writer = csv.writer(csvoutput, lineterminator='\n') 
    reader = csv.reader(csvinput) 

    all = [] 
    row = next(reader) 
    row.append('FileName') 
    all.append(row) 

    for row in reader: 
     row.append(row[1]) 
     all.append(row) 

    writer.writerows(all) 

Answer 1

我不知道如果我完全理解你的問題，但在最壞的情況下這應該給你一些想法如何解決你的問題。

import csv 
import os 

for file_name in os.listdir(): 
    with open(file_name,'r') as csvinput: 
     reader = csv.reader(csvinput) 

     all = [] 
     row = next(reader) 
     row.append('FileName') 
     all.append(row) 

     for row in reader: 
      row.append(file_name) 
      all.append(row) 

     with open(file_name, 'w') as csvoutput: 
      writer = csv.writer(csvoutput, lineterminator='\n') 
      writer.writerows(all)