在 Python 的 matplotlib 中向轴添加彩色框

Adding colored boxes to axes in Python's matplotlib

社区,

说我用 Python 的 matplotlib 创建了一个散点图:

plt.scatter(x, y)

假设这是数据:

现在,有没有办法像这样在轴上添加彩色框(在给定的 x 值之间,例如:从 x=-0.2 到 x=0 添加一个绿色框,添加一个 ...) :

包括文本标签(我猜是在中间范围内)。

老实说,我什至不确定如何开始这个(除了制作散点图)。

问题 谁能至少指导我执行此操作的 matplotlib 功能(或任何其他 Python 包)?

非常感谢您的帮助。

fig = plt.figure(figsize=(6,4))
plt.plot(np.arange(0,7,1),np.arange(0,7,1),linestyle='--', marker='o', color='b')
rect1 = plt.Rectangle((1.5,1.5),(0.7),(0.5), facecolor="grey",edgecolor="black",alpha=0.8)
rect2 = plt.Rectangle((2.5,2.5),(0.7),(0.5), facecolor="yellow",edgecolor="black",alpha=0.8)
rect3 = plt.Rectangle((3.5,3.5),(0.7),(0.5), facecolor="k",edgecolor="black",alpha=0.8)
plt.gca().add_patch(rect1)
plt.gca().add_patch(rect2)
plt.gca().add_patch(rect3)

plt.text(1.65,1.65,r"Grey",fontsize = 10,zorder = 5,color = 'k',fontweight = 'bold')
plt.text(2.55,2.65,r"Yellow",fontsize = 10,zorder = 5,color = 'k',fontweight = 'bold')
plt.text(3.555,3.65,r"Black",fontsize = 10,zorder = 5,color = 'white',fontweight = 'bold')  

您可能想使用 matplotlib 的 text() 功能。也许是这样的,

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 1, 100)
y = np.sin(2*np.pi * x)

fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot(x, y, 'o')
ax.set_xlim(-0.2, 1.2)
ax.set_ylim(-1.5, 1.5)

x_ticks = ax.get_xticks()
y_ticks = ax.get_yticks()
# use len(x_ticks)-1 number of colors
colors = ['b', 'g', 'r', 'c', 'm', 'y', 'orange']
for i, x_tick in enumerate(x_ticks[:-1]):
    ax.text(x_tick+0.03, y_ticks[0]-0.165, "text %i"%i,
            bbox={'fc':colors[i], 'pad':10})

这个代码returns这个image。您可以根据需要调整填充和 x,y 位置以获得您想要的精确外观。