您可以使用usecols
參數來篩選您希望提取的數據列。通過在讀取數據時指定usecols
參數,您可以只選擇需要的列,而不必加載整個數據集。這樣可以節省內存和加快數據處理速度。
以下是使用usecols
參數來篩選數據列的示例代碼:
import pandas as pd
# 讀取數據,只選擇需要的列
data = pd.read_csv('data.csv', usecols=['column1', 'column2'])
# 顯示數據的前幾行
print(data.head())
在上面的示例中,usecols
參數被設置為['column1', 'column2']
,這意味著只選擇data.csv
文件中的column1
和column2
列。您可以根據需要指定任意數量的列。