Python3 寫入CSV文件出現空餘行

版權聲明：本文爲博主原創文章，未經博主允許不得轉載。 https://blog.csdn.net/PinkRiverside/article/details/81388082

原代碼：

import csv
import os

# 2016-01-01 週五
i = 4

with open(os.path.join(os.path.dirname(__file__), "data/weekend.csv")) as csvFile:
    rows = csv.reader(csvFile)
    with open(os.path.join(os.path.dirname(__file__), "data/test.csv"), 'w') as f:
        writer = csv.writer(f)
        for row in rows:
            row=list(row)
            if i > 5:
                row.append("1") # 週末爲1
            else:
                row.append("0") # 工作日爲0
            i = i % 7 + 1
            writer.writerow(row)

目的是將從2016-01-01至2018-05-31的日期貼上標籤，工作日爲0，週末爲1。代碼運行後結果如下，會帶有空餘行：

解決方案：在open的時候加上 newline=”” 這個參數。
修改後的代碼：

import csv
import os

# 2016-01-01 週五
i = 4

with open(os.path.join(os.path.dirname(__file__), "data/weekend.csv")) as csvFile:
    rows = csv.reader(csvFile)
    with open(os.path.join(os.path.dirname(__file__), "data/test.csv"), 'w',newline='') as f:
        writer = csv.writer(f)
        for row in rows:
            row=list(row)
            if i > 5:
                row.append("1") # 週末爲1
            else:
                row.append("0") # 工作日爲0
            i = i % 7 + 1
            writer.writerow(row)

修改後運行結果：