Как установить xlim и ylim для подзадачи в matplotlib

Question

Я хотел бы ограничить оси X и Y в matplotlib, но для специального подзаголовка. Как я вижу, сама фигура подзаголовка не имеет никакого свойства оси. Я хочу, например, изменить только лимиты для второго сюжета!

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

Answer 1

Вы должны изучить немного интерфейса OO для matplotlib, а не только интерфейс конечной машины. Почти все функции 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)