sklearn数据集中,如何根据特征名读取一列数据。

boston = load_boston()
data=boston.data
print(boston.feature_names)

显示特征名:['CRIM' 'ZN' 'INDUS' 'CHAS' 'NOX' 'RM' 'AGE' 'DIS' 'RAD' 'TAX' 'PTRATIO'
'B' 'LSTAT']

如何根据特征名读取一列数据出来呢。

以第一列特征为例:data["CRIM"],就可以读取CRIM这一列的特征数据了

结合pandas就可以直接读取:

from sklearn.datasets import load_boston
import pandas as pd


boston = load_boston()
 #把波士顿数据集的数据和特征名分别写入DataFrame中,然后可以直接读取指定的某一列数据
df = pd.DataFrame(data=boston.data, columns=boston.feature_names)
df['LSTAT']