使用pandas的read_csv函數(shù)讀取csv文件，并通過指定columns參數(shù)來選擇需要讀取的列。
示例代碼如下：
import pandas as pd
# 讀取整個(gè)csv文件，不指定列
df = pd.read_csv('data.csv')
# 讀取指定列
selected_columns = ['column1', 'column2', 'column3']
df_selected = pd.read_csv('data.csv', usecols=selected_columns)
在第二個(gè)例子中，使用了usecols參數(shù)來指定需要讀取的列，將列名以列表形式傳遞給該參數(shù)。
需要注意的是，如果要讀取的csv文件中的列名與selected_columns中的列名不完全匹配，可能會(huì)出現(xiàn)KeyError錯(cuò)誤。因此，確保列名的正確性非常重要。
億速云公眾號(hào)
手機(jī)網(wǎng)站二維碼