用python写一个2D数据plt.colorbar

好的，以下是一个例子：

import matplotlib.pyplot as plt
import numpy as np

# 生成2D数据
data = np.random.rand(10, 10)

# 绘制热力图
plt.imshow(data)

# 添加colorbar
plt.colorbar()

# 显示图像
plt.show()

这个例子中，我们首先生成了一个 $10 \times 10$ 的随机数组，然后使用 `imshow` 函数绘制了一个热力图。接着使用 `colorbar` 函数添加了一个颜色条。最后调用 `show` 函数显示图像。