matplotlib柱状图显示条件颜色
时间: 2023-10-03 13:09:37 浏览: 182
颜色渐变的柱状图
3星 · 编辑精心推荐
你可以使用 `bar` 函数绘制柱状图,并使用 `color` 参数指定颜色。要根据条件动态更改颜色,可以使用循环遍历每个柱子,并按条件设置相应的颜色。
下面是一个示例代码,根据柱子高度设置颜色:
```python
import matplotlib.pyplot as plt
import numpy as np
# 生成数据
x = np.arange(5)
y = np.array([3, 8, 4, 2, 5])
# 颜色条件
threshold = 4
# 循环遍历每个柱子,根据条件设置颜色
colors = []
for i in range(len(y)):
if y[i] >= threshold:
colors.append('r')
else:
colors.append('b')
# 绘制柱状图
plt.bar(x, y, color=colors)
plt.show()
```
运行代码后,你会看到高于阈值 `4` 的柱子被标记为红色,低于阈值的柱子被标记为蓝色。你可以根据自己的需求修改阈值和颜色条件。
阅读全文