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

Barbara Streisand
發布: 2024-10-31 18:28:29
原創
516 人瀏覽過

How to Exclude Headers When Processing CSV Files with Python?

使用 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中文網其他相關文章!

來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
作者最新文章
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板