用python做因素重要性后,怎样按数据大小排名做横向的柱状图
时间: 2023-06-17 10:04:24 浏览: 52
首先,你需要将因素重要性的结果存储为一个 pandas DataFrame,假设它的列名为 "feature" 和 "importance",可以按以下步骤进行排序并绘制横向柱状图:
```python
import pandas as pd
import matplotlib.pyplot as plt
# 读取因素重要性结果
df_importance = pd.read_csv("importance.csv")
# 按重要性大小降序排序
df_importance = df_importance.sort_values("importance", ascending=False)
# 绘制横向柱状图
fig, ax = plt.subplots(figsize=(8, 6))
ax.barh(df_importance["feature"], df_importance["importance"])
ax.invert_yaxis() # 反转 y 轴,使得重要性高的因素在上方
ax.set_xlabel("Importance")
ax.set_title("Feature Importance")
plt.show()
```
这段代码会生成一个横向的柱状图,其中 x 轴表示因素重要性,y 轴表示因素名称,因素名称按照重要性从高到低排列。你可以根据需要调整图形大小、颜色、字体等参数。