如何使用 Python 從 CSV 檔案中提取特定列？

使用 Python csv 模組修剪 CSV 欄位

通常，需要解析逗號號分隔值 (CSV) 檔案並擷取特定的資料列。使用Python的csv模組，可以輕鬆完成這項任務。

列提取

要提取特定列，請使用included_cols變量，該變量將列索引指定為包括。每列從 0 開始編號，第一列為 0。例如，要擷取「ID」、「名稱」、「郵遞區號」和「電話」列，您可以編寫Included_cols = [1, 2, 6 , 7].

迭代和列印

要迭代行並提取指定列，請在for 循環中使用列表理解。對於每一行，內容變數將使用指定列中的值進行填入。若要顯示擷取的數據，請在迴圈中使用 print(content)。

included_cols = [1, 2, 6, 7]

for row in reader:
    content = list(row[i] for i in included_cols)
    print(content)

使用 Pandas 進行最佳化

為了更有效率地處理 CSV，請考慮使用 pandas 模組。 Pandas 提供了方便且強大的工具來讀取和操作 CSV 檔案。要讀取 CSV 並將特定列儲存到變數中，只需使用：

import pandas as pd
df = pd.read_csv(csv_file)
saved_column = df.column_name

以上是如何使用 Python 從 CSV 檔案中提取特定列？的詳細內容。更多資訊請關注PHP中文網其他相關文章！