Python colorbar

在 Python 中，我们可以使用 matplotlib 库来创建颜色条（colorbar）。下面是一个简单的例子，演示如何创建一个颜色条：

import matplotlib.pyplot as plt
import numpy as np

# 创建一个 2D 数组
data = np.random.rand(10, 10)

# 绘制图像
plt.imshow(data)

# 添加颜色条
plt.colorbar()

# 显示图像
plt.show()

在上面的例子中，我们首先创建了一个 10x10 的随机数组，并使用 `imshow` 函数将其绘制为图像。接着，我们使用 `colorbar` 函数添加了一个颜色条。最后，我们使用 `show` 函数显示了图像和颜色条。

如果你想更改颜色条的颜色和范围，可以使用 `cmap` 和 `vmin`、`vmax` 参数。例如：

import matplotlib.pyplot as plt
import numpy as np

# 创建一个 2D 数组
data = np.random.rand(10, 10)

# 绘制图像
plt.imshow(data, cmap='coolwarm', vmin=0, vmax=1)

# 添加颜色条
plt.colorbar()

# 显示图像
plt.show()

在上面的例子中，我们使用了 `cmap` 参数来指定颜色条的颜色映射，使用 `vmin` 和 `vmax` 参数来指定颜色条的范围。