如果条件为真,如何更改单个条形的颜色(Matplotlib)?

matplotlibpythondata visualization

要更改单个条形的颜色(如果条件为真),我们可以创建一组值和一个颜色列表,其中红色直到值为 2;否则在列表中添加黄色。

步骤

  • 设置图形大小并调整子图之间和周围的填充。
  • 初始化条形的可变宽度。
  • 创建两个值和颜色列表。
  • 使用 bar() 方法绘制条形图。
  • 要显示图形,请使用 show() 方法。

示例

import numpy as np
import matplotlib.pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
data = np.arange(5)
width = 0.5
vals = [1, 2, 1, 5, 3]
colors = ["red" if i != 2 else "yellow" for i in vals]
plt.bar(data, vals, width, color=colors)
plt.show()

输出


相关文章