使用plt.bar()画垂直条形图怎么让条形图颜色盘系统自定义非用户指定?
时间: 2024-02-01 15:16:09 浏览: 103
python画图-使用Python实现的数据可视化画图之条形图示例.zip
可以使用 `plt.cm` 模块中的颜色映射函数,如 `plt.cm.viridis`、`plt.cm.coolwarm` 等,这些函数可以生成一组颜色,并且可以根据数据的大小自动调整颜色深浅。在调用 `plt.bar()` 方法时,可以将 `color` 参数设置为颜色映射函数的返回值。
以下是一个简单的示例代码:
```python
import numpy as np
import matplotlib.pyplot as plt
# 生成示例数据
x = np.arange(5)
y = np.random.randint(1, 10, size=5)
# 定义颜色映射函数
cmap = plt.cm.viridis
# 使用 plt.bar() 绘制垂直条形图,并设置颜色
plt.bar(x, y, color=cmap(x))
# 显示图像
plt.show()
```
在上述代码中,我们通过 `cmap(x)` 来获取颜色映射函数返回的一组颜色,其中 `x` 是条形图的 x 坐标轴数据。这样就可以让条形图的颜色使用颜色映射函数自定义的颜色了。
阅读全文