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

Question

社区，

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

plt.scatter(x, y)

假设这是数据：

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

包括文本标签（我猜是中档）。

说实话，我什至不知道如何开始这个（除了制作散点图）。

问题 谁能至少指导我使用 matplotlib 的一个功能（或任何其他 Python 包）？

非常感谢您的帮助。

Answer 1

您可能想使用 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})

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

Answer 2

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