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