您可以使用以下两种方法来选取指定列:
使用列名选取:
import pandas as pd df = pd.read_csv('data.csv') col_data = df[['column_name1', 'column_name2']] print(col_data)
使用iloc和loc选取:
# 使用iloc选取第一列和第二列 col_data = df.iloc[:, [0, 1]] # 使用loc选取第一行和第二行,以及第一列和第二列 col_data = df.loc[0:1, ['column_name1', 'column_name2']]