使用pandas的read_csv函數讀取csv文件,并通過指定columns參數來選擇需要讀取的列。
示例代碼如下:
import pandas as pd
# 讀取整個csv文件,不指定列
df = pd.read_csv('data.csv')
# 讀取指定列
selected_columns = ['column1', 'column2', 'column3']
df_selected = pd.read_csv('data.csv', usecols=selected_columns)
在第二個例子中,使用了usecols參數來指定需要讀取的列,將列名以列表形式傳遞給該參數。
需要注意的是,如果要讀取的csv文件中的列名與selected_columns中的列名不完全匹配,可能會出現KeyError錯誤。因此,確保列名的正確性非常重要。