python - 使用 Matplotlib 在一张条形图中绘制两个字典

Question

这个问题已经有答案了:

How to plot a grouped bar chart from multiple datasets (3 个回答)

已关闭 5 年前。

我有这两个字典，我想使用 Matplotlib 在同一个条形图中绘制它们:

accuracy_pre = {"Random Forest": rf_accuracy_score, "Logistic Regression": log_accuracy_score, "KNN": knn_accuracy_score}

accuracy_post = {"Random Forest": gs_rf_accuracy_score, "Logistic Regression": gs_logmodel_accuracy_score, "KNN": knn_accuracy_score_iter}

字典值是整型变量。两个字典都有相同的键，因此总共有 6 个小节，但属于相同键的小节将彼此相邻。我可以创建两个单独的条形图，但无法将它们放在一起。有人可以帮助我吗？

这是我已有的条形图的代码:

X = np.arange(len(accuracy_pre))
plt.bar(X, accuracy_pre.values(), align='center', width=0.5)
plt.xticks(X, accuracy_pre.keys()) 
plt.title("Accuracy score - before grid search", fontsize=17)
plt.ylim(0, 1)

Answer 1

我稍微调整了你的代码以获得所需的情节。希望这有帮助!

import numpy as np
import matplotlib.pyplot as plt

#sample data
accuracy_pre = {"Random Forest": 1, "Logistic Regression": 2, "KNN": 3}
accuracy_post = {"Random Forest": 4, "Logistic Regression": 5, "KNN": 6}

X = np.arange(len(accuracy_pre))
ax = plt.subplot(111)
ax.bar(X, accuracy_pre.values(), width=0.2, color='b', align='center')
ax.bar(X-0.2, accuracy_post.values(), width=0.2, color='g', align='center')
ax.legend(('Pre Accuracy','Post Accuracy'))
plt.xticks(X, accuracy_pre.keys())
plt.title("Accuracy score", fontsize=17)
plt.show()