python pandas 根据列名取数据

pandas中DataFrame想根据列名读取数据，数据如下图

想获取feature_name: ['DVI', 'NDVI_B8A7', 'NDVI_B86']的数据

代码如下：

#从excel中读取数据
dataset_path=r'C:\Users\Administrator\Desktop\test\sample.xlsx' 
x_train = pd.read_excel(dataset_path,sheet_name= 'x_train')
train_data = x_train[feature_name] 
print('train_data:',train_data.head(5))

 结果如下：