Построение двух фигур рядом

Question

Как видно из названия, я изо всех сил пытаюсь построить два сюжета вместе, бок о бок. Концептуально, код выглядит следующим образом:

def my_func(arr):    
    plt.scatter(arr[:, 0], arr[:, 1])

fig, ax = plt.subplots(1, 2, sharex='col', sharey='row')

arr1 = np.array([[1, 2], [2, 2], [4, 3], [6, 4], [5, 6]])

for i in range(2):
   my_func(arr1 + i)

Проблема здесь в том, чтобы построить два графика вместе, используя my_fun c - функцию, которая создает график (используя несколько параметров, поэтому она должна быть отдельная функция). Проблема здесь заключается в том, что два графика, которые должны были быть нанесены в два разных блока, отображаются в одном блоке. Как это исправить?

Answer 1

Вам необходимо передать соответствующие объекты оси в вашу функцию для построения

def my_func(arr, ax):    
    ax.scatter(arr[:, 0], arr[:, 1])

fig, ax = plt.subplots(1, 2, sharex='col', sharey='row')

arr1 = np.array([[1, 2], [2, 2], [4, 3], [6, 4], [5, 6]])

for i in range(2):
   my_func(arr1 + i, ax[i])