處理包含標頭的CSV(逗號分隔值)檔案時,通常需要在處理過程中排除這些標頭。本文解決了嘗試在 Python 中跳過標題時遇到的常見問題。
提供的程式碼片段遇到標題行受應用程式影響的問題。為了修正這個問題,讀者應該注意到 reader 變數會迭代 CSV 檔案中的行。
要在主循環之前跳過一行(其中行索引從1 開始),請使用next() 函數,如下所示:
next(reader, None) # Skip header by returning None if the reader is empty
此外,為了增強可讀性並簡化文件處理,可以使用上下文管理器:
with open("tmob_notcleaned.csv", "rb") as infile: with open("tmob_cleaned.csv", "wb") as outfile: reader = csv.reader(infile) next(reader, None) # Skip headers writer = csv.writer(outfile) for row in reader: # Process rows here
或者,要在輸出文件中包含標題行,只需傳遞headers 變量,可以使用next() 初始化,給作者:
headers = next(reader, None) # Get headers or None if empty if headers: writer.writerow(headers)
透過遵循這些技術,開發人員可以有效地跳過標題並輕鬆處理CSV 檔案。
以上是在 Python 中處理 CSV 檔案時如何跳過標頭?的詳細內容。更多資訊請關注PHP中文網其他相關文章!