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

Question

社区，

假设我使用Python的matplotlib创建了一个散点图：

plt.scatter(x, y)

让我们说这是数据：

现在，有没有办法在轴上添加彩色框（在给定的x值之间，例如：添加一个绿色框，从x = -0.2到x = 0，添加一个......），如下所示：

包括文字标签（我猜中间范围）。

我甚至不确定如何开始这个问题，说实话（除了制作散点图）。

问题 任何人都可以至少指导我使用matplotlib的功能（或任何其他Python包）吗？

非常感谢您的帮助。

Answer 1

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')  

Answer 2

你可能想要使用matplotlib＆＃39; s 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})

此代码返回此image。您可以根据需要调整填充和x，y位置，以获得您想要的精确外观。