如何使用 CSV 模組從 CSV 檔案中的特定列中提取資料？

使用CSV 模組從CSV 檔案中的特定列提取資料

在處理CSV 檔案時，經常需要擷取資料僅來自特定列。要使用 CSV 模組實現此目的，了解該程式庫如何處理資料檢索非常重要。

在提供的程式碼範例中，嘗試透過 row[i] 語法存取特定列，其中 i 代表列號。然而，由於方法不正確，結果並沒有達到預期。

正確的方法是將 print 語句包含在 for 迴圈中。以下更正後的程式碼示範了這一點：

import csv

included_cols = [1, 2, 6, 7]
with open("csv_file.csv") as csvfile:
    reader = csv.reader(csvfile, delimiter=",")
    for row in reader:
        content = list(row[i] for i in included_cols)
        print(content)  # Include the print statement within the loop to display each row

現在，修改後的程式碼將列印出每行所需列中的值。

但是，值得一提的是使用以下方法的替代方法： pandas 函式庫，簡化了 CSV 檔案的處理。使用pandas，您可以輕鬆讀取CSV 檔案並將特定列提取到變數中：

import pandas as pd

df = pd.read_csv("csv_file.csv")
names = df["Name"]  # Use df['column_name'] to extract a specific column into a variable

pandas 庫提供了一組用於處理結構化資料的多功能工具，使其成為處理CSV文件的推薦選擇有效。

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