Sci Kit Learn Y ticks Matplotlib

时间:2018-03-13 16:13:52

标签: python matplotlib machine-learning scikit-learn data-science

我使用Sci Kit Learns网站上的代码进行回归算法,然后绘制Deviance& amp; matplotlib的变量重要性。

[Sci Kit学习梯度升压回归] [1]

Sci Kit了解着名的波士顿住房项目数据集的示例,该数据集在绘制数据时内置于Sci Kit学习包中,matplotlib引用boston.feature_names,我试图通过我的Pandas数据框找到它。

plt.yticks(pos, boston.feature_names[sorted_idx])

在我的Pandas数据框中,是否可以将Y刻度改为我的列标题名称?

1 个答案:

答案 0 :(得分:0)

这将引用pandas数据帧列名:

plt.yticks(pos, df.columns.values)