python colorbar

作者：战虎天下发布日期:2026-03-29 浏览:65

import matplotlib.pyplot as plt
import numpy as np

# 创建一个示例数据集
data = np.random.rand(10, 10)

# 创建一个新的图形
plt.figure()

# 使用imshow绘制热图，并添加colorbar
cax = plt.imshow(data, cmap='viridis')
plt.colorbar(cax)

# 添加标题和标签
plt.title('Heatmap with Colorbar')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')

# 显示图形
plt.show()

解释说明：

numpy 用于生成一个随机的二维数组 data，作为热图的数据源。
matplotlib.pyplot 是用来绘制图像的主要库。
plt.imshow() 函数用于绘制热图，cmap='viridis' 指定了颜色映射方案（可以替换为其他颜色映射，如 'plasma', 'magma', 'inferno' 等）。
plt.colorbar() 函数用于在图像旁边添加一个颜色条（colorbar），表示颜色与数值之间的对应关系。
plt.title(), plt.xlabel(), 和 plt.ylabel() 分别用于设置图像的标题和坐标轴标签。
最后，plt.show() 用于显示绘制的图像。

上一篇：python json load

下一篇：pythonweb开发