首頁 > 後端開發 > Python教學 > 如何有效率地讀取CSV檔案中的特定欄位?

如何有效率地讀取CSV檔案中的特定欄位?

Patricia Arquette
發布: 2024-11-22 10:32:10
原創
584 人瀏覽過

How to Read Specific Columns from a CSV File Efficiently?

使用csv 模組選擇性讀取CSV 檔案中的欄位

理解問題

從CSV 檔案中讀取特定欄位可能是一項常見任務在資料處理中。但是,按照給定程式碼中的嘗試按列號訪問列可能會導致意外結果。

解決錯誤

所提供程式碼中的錯誤在於將 print 語句放置在for 迴圈。因此,僅列印最後一次迭代的結果,即 CSV 檔案的最後一列。

程式碼修正

要解決此問題,應將print 語句移到迴圈內進行列印每行的具體欄位:

for row in reader:
    content = list(row[i] for i in included_cols)
    print(content)
登入後複製

Pandas 模組的優點

而調整後的程式碼可以完成任務,缺乏pandas模組的便利性和效率。 Pandas 為 CSV 檔案操作提供了優雅且多功能的解決方案。

用於選擇性列讀取的Pandas

使用pandas,從CSV 檔案中讀取特定欄位變得毫不費力:

import pandas as pd

df = pd.read_csv(csv_file)

# Save column data to a variable
column_data = df['column_name']
登入後複製

簡化的程式碼

Pandas 解顯著簡化了程式碼:

import pandas as pd

df = pd.read_csv(csv_file)

# Save specific columns as a list
specific_columns = ['ID', 'Name', 'Zip', 'Phone']
data = df[specific_columns].values.tolist()
登入後複製

以上是如何有效率地讀取CSV檔案中的特定欄位?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

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