改进图表
为让图表更易读,我们需要达成两个目标:
- 按升序重新排列柱形顺序。
- 为图表添加与特征名称对应的标签。
为此,我们将利用 NumPy 的索引功能。.argsort() 方法会对数组排序并返回索引。我们将使用这些索引来同时实现这两个目标。
本练习是课程的一部分
营销分析:用 Python 预测客户流失
练习说明
- 使用
np.argsort()作用于importances,计算其排序后的索引sorted_index。 - 通过访问
X的列并用sorted_index进行索引,创建排序后的标签。 - 通过使用
sorted_index对importances进行索引并指定关键字参数tick_label=labels来创建图表。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Sort importances
sorted_index = ____(____)
# Create labels
labels = X.columns[____]
# Clear current plot
plt.clf()
# Create plot
plt.barh(range(X.shape[1]), importances[____], tick_label=____)
plt.show()