python中热力图颜色变化代码
时间: 2023-09-24 08:13:42 浏览: 159
python绘制热力图利用python中的matplotlib中的pyplot包绘制二维热力图
5星 · 资源好评率100%
可以使用matplotlib库中的imshow函数来显示热力图,并且可以通过cmap参数指定颜色映射。以下是一个简单的例子:
```python
import matplotlib.pyplot as plt
import numpy as np
# 生成随机数作为数据
data = np.random.rand(10, 10)
# 绘制热力图,使用绿到红的颜色映射
plt.imshow(data, cmap='RdYlGn')
# 显示颜色条
plt.colorbar()
# 显示图像
plt.show()
```
上述代码中,生成了一个10x10的随机矩阵作为数据,并使用'RdYlGn'颜色映射绘制热力图。可以通过更改cmap参数来使用其他的颜色映射。另外,可以使用colorbar函数来添加颜色条。
阅读全文