使用 Python 處理 CSV 檔案時如何排除標頭？

使用 Python 處理 CSV 檔案時排除標頭

處理 CSV 檔案時，可能需要從某些操作中排除標頭。這可以透過利用 Python 的 CSV 讀取器和寫入器模組來實現。

在提供的程式碼中，需要跳過標題行。一種更直接的方法是在處理其餘行之前跳過第一行，而不是將行變數初始化為 1。這可以如下完成：

<code class="python">with open("tmob_notcleaned.csv", "rb") as infile, open("tmob_cleaned.csv", "wb") as outfile:
    reader = csv.reader(infile)
    next(reader, None)  # Skip the headers
    writer = csv.writer(outfile)
    for row in reader:
        # Process each row
        writer.writerow(row)</code>

透過呼叫 next(reader, None)，擷取並丟棄 CSV 檔案的第一行。然後可以處理以下行並將其寫入輸出檔案。

此外，可以透過使用上下文管理器自動處理文件開啟和關閉來簡化程式碼：

<code class="python">with open("tmob_notcleaned.csv", "rb") as infile, open("tmob_cleaned.csv", "wb") as outfile:
    reader = csv.reader(infile)
    writer = csv.writer(outfile)
    # Skip the headers
    if headers := next(reader, None):
        writer.writerow(headers)</code>

在此程式碼中，可選headers 變數接收跳過的標頭，允許使用者根據需要將未處理的標頭寫入輸出檔案。

以上是使用 Python 處理 CSV 檔案時如何排除標頭？的詳細內容。更多資訊請關注PHP中文網其他相關文章！