1. 程式人生 > 其它 >Python 合併多個csv檔案,並將檔名寫入新的列

Python 合併多個csv檔案,並將檔名寫入新的列

技術標籤:pythonpythoncsv

readList = []  #用來裝合併檔案的內容
titleList = ["date","district","region"] #檔案標題

for i in listName:
    #i 是檔名
#############讀取多個csv檔案內容###################
    with open('{}.csv'.format(i),'r',newline="",encoding="GB18030") as read_csvfile:
            readcsv_all = csv.reader(read_csvfile)
            next(read_csvfile)        #跳過第一行“標題”
            for line in readcsv_all:
                line.append(i)        #新增“檔名”作為新的列
                readList.append(line) 

############寫入檔案##############################
with open('alldata.csv', 'a+' , newline = "" , encoding = "GB18030") as f:
        csv_write = csv.writer(f)
        csv_write.writerow(titleList)  #寫入檔案標題
        for line in readList:
#             print(line)
            csv_write.writerow(line)