如何在matplotlib中为子图设置xlim和ylim

时间:2013-04-07 02:20:55

标签: python matplotlib plot

我想限制matplotlib中的X和Y轴,但是对于特定的子图。我所看到的 subplot figure本身没有任何axis属性。我想要例如只更改第二个图的限制!

import matplotlib.pyplot as plt
fig=plt.subplot(131)
plt.scatter([1,2],[3,4])
fig=plt.subplot(132)
plt.scatter([10,20],[30,40])
fig=plt.subplot(133)
plt.scatter([15,23],[35,43])
plt.show()

1 个答案:

答案 0 :(得分:194)

你应该学习matplotlib的一些OO接口,而不仅仅是状态机接口。几乎所有plt.*函数都是瘦包装器,基本上都是gca().*

plt.subplot会返回axes个对象。一旦你有了对象的参考,你就可以直接绘制它,改变它的极限等等。

import matplotlib.pyplot as plt

ax1 = plt.subplot(131)
ax1.scatter([1, 2], [3, 4])
ax1.set_xlim([0, 5])
ax1.set_ylim([0, 5])


ax2 = plt.subplot(132)
ax2.scatter([1, 2],[3, 4])
ax2.set_xlim([0, 5])
ax2.set_ylim([0, 5])

等等,可以根据需要设置多个轴。

或者更好,将它们全部包装成循环:

import matplotlib.pyplot as plt

DATA_x = ([1, 2],
          [2, 3],
          [3, 4])

DATA_y = DATA_x[::-1]

XLIMS = [[0, 10]] * 3
YLIMS = [[0, 10]] * 3

for j, (x, y, xlim, ylim) in enumerate(zip(DATA_x, DATA_y, XLIMS, YLIMS)):
    ax = plt.subplot(1, 3, j + 1)
    ax.scatter(x, y)
    ax.set_xlim(xlim)
    ax.set_ylim(ylim)