Как накладывать фигуры в matplotlib

Question

Я создаю несколько рисунков в нескольких модулях, и я бы хотел наложить их на мой main.py.

Могу ли я вернуть полезную фигуру/график, который можно было бы повторно использовать в main.py, чтобы создать комбинированный рисунок?

Спасибо!

Answer 1

Решение состоит в том, чтобы установить фигуру/ось в main.py, а затем передать рукоятку оси каждому модулю. В качестве минимального примера

import matplotlib.pyplot as plt 
import numpy as np 
#from mod import plotsomefunction 
#from diffrentmod import plotsomeotherfunction 

def plotsomefunction(ax, x): 

    return ax.plot(x, np.sin(x)) 

def plotsomeotherfunction(ax, x): 

    return ax.plot(x,np.cos(x)) 


fig, ax = plt.subplots(1,1) 
x = np.linspace(0,np.pi,1000) 
l1 = plotsomefunction(ax, x) 
l2 = plotsomeotherfunction(ax, x) 
plt.show() 

где функции представляют собой модули.

В качестве альтернативы вы можете просто создать фигуру в главном окне и добавить ее к текущей оси в каждом модуле с помощью plt.sca. Это кажется гораздо менее надежным решением.