python使用matplotlib绘制水平条形图并在条形图上添加实际数值标签实战

Posted 2021-09-23 Data+Science+Insight

python使用matplotlib绘制水平条形图并在条形图上添加实际数值标签实战

# 绘制水平条形图进行模型对比

x_labels = ["LogisticRegression", "NaiveBayesian",\'SVM\',\'RandomForest\',\'Xgboost\',\'LightGBM\']
prediction_acc = [0.8,0.9,0.77,0.56,0.99,0.79]

import matplotlib.pyplot as plt


n_bars = len(x_labels)
xval = np.arange(n_bars)

colors = [\'r\', \'g\', \'b\', \'orange\', \'black\',\'purple\']

# plot diabetes results
plt.figure(figsize=(10, 7))
ax1 = plt.subplot(111)
for j in xval:
    ax1.barh(j, prediction_acc[j],
             color=colors[j], alpha=0.6, align=\'center\')

for i, v in enumerate(predict