Matplotlib - 设置特定象限的背景颜色

时间:2013-04-24 04:48:45

标签: python matplotlib

Jfreechart中有一个名为setQuadrantPaint的方法,您可以在图中设置给定quandrant的背景颜色。

您如何在matplotlib中获得相应的效果?

E.g。

enter image description here

1 个答案:

答案 0 :(得分:3)

您可以在后台使用imshow绘制2x2数组。赋予它一定程度将使其中心始终为0,0。

import numpy as np
import matplotlib.pyplot as plt

x1, y1 = np.random.randint(-8,8,5), np.random.randint(-8,8,5)
x2, y2 = np.random.randint(-8,8,5), np.random.randint(-8,8,5)

vmax = np.abs(np.concatenate([x1,x2,y1,y2])).max() + 5

extent = [vmax*-1,vmax, vmax*-1,vmax]
arr = np.array([[1,0],[0,1]])


fig, ax = plt.subplots(1,1)

ax.scatter(x1,y1, marker='s', s=30, c='r', edgecolors='red', lw=1)
ax.scatter(x2,y2, marker='s', s=30, c='none', edgecolors='red', lw=1)

ax.autoscale(False)
ax.imshow(arr, extent=extent, cmap=plt.cm.Greys, interpolation='none', alpha=.1)

ax.axhline(0, color='grey')
ax.grid(True)

enter image description here

在绘制数据点之后,但在图像出现之前,将自动缩放设置为False,确保轴仅缩放到数据点。