如何使用matplotlib python添加色彩映射和矩形框？

import matplotlib.pyplot as plt
import numpy as np

# 生成一些示例数据
x = np.random.rand(100)
y = np.random.rand(100)
colors = np.random.rand(100)

# 绘制散点图，并使用色彩映射
plt.scatter(x, y, c=colors, cmap='viridis')

# 添加颜色条
plt.colorbar()

# 显示图形
plt.show()

import matplotlib.pyplot as plt
from matplotlib.patches import Rectangle

# 创建一个矩形对象
rect = Rectangle((0.2, 0.3), 0.4, 0.5, edgecolor='red', facecolor='none')

# 创建一个图形对象
fig, ax = plt.subplots()

# 将矩形对象添加到图形中
ax.add_patch(rect)

# 设置坐标轴范围
ax.set_xlim(0, 1)
ax.set_ylim(0, 1)

# 显示图形
plt.show()