使用Python 處理CSV 檔案時跳過標頭
問題:
試試時使用Python 的CSV 檔案時,使用者會遇到一個問題,其中包含標題的第一行正在被處理函數修改。目標是從第二行開始編輯 CSV 文件,不包括標題。
解決方案:
要解決此問題,可以使用由Python CSV 模組可用於在處理過程中跳過標頭。以下步驟概述了解決方案:
將CSV 檔案作為上下文管理器開啟:
<code class="python">with open("tmob_notcleaned.csv", "rb") as in_file, open("tmob_cleaned.csv", "wb") as out_file:</code>
使用with 作為上下文管理器自動處理開啟和關閉CSV 文件,確保正確的資源管理。
建立Reader 與Writer 物件:
<code class="python">reader = csv.reader(in_file) writer = csv.writer(out_file)</code>
csv .reader 建立一個可迭代物件來迭代CSV 行,而csv.writer允許將行寫入輸出CSV 檔案。
跳過標題:
<code class="python">next(reader, None)</code>
此行將迭代器前進到第一行,而不將其分配給變數。透過提供 None 作為第二個參數,跳過的行將被丟棄。
處理並寫入剩餘行:
<code class="python">for row in reader: # Perform processing writer.writerow(row)</code>
迭代其餘行(不包括標題)執行必要的處理,並將修改後的行寫入輸出檔。
可選擇寫入標題(如果需要):
<code class="python">headers = next(reader, None) if headers: writer.writerow(headers)</code>
此程式碼區塊允許透過將next() 的結果傳遞給writer.writerow() 將標頭寫入未處理的輸出檔。
By依照這些步驟,Python 程式碼在處理 CSV 檔案時會跳過標頭,確保第一行保持不變。
以上是使用 Python 處理 CSV 檔案時如何跳過標頭?的詳細內容。更多資訊請關注PHP中文網其他相關文章!