befüllen Matplotlib Subplots durch eine Schleife und eine Funktion

Question

Ich muss Unterschritte einer Figur durch Schleifen Iterationen zeichnen; Jede Iteration ruft eine Funktion auf, die in einem anderen Modul definiert ist (= eine andere Py-Datei), die ein Paar Unterplots zeichnet. Hier ist, was ich versuchte - und leider nicht funktioniert:

1) Vor der Schleife, erstellen Sie eine Figur mit der ausreichenden Anzahl von Zeilen und 2 Spalten:

import matplotlib.pyplot as plt  
fig, axarr = plt.subplots(nber_rows,2) 

2) Innerhalb der Schleife bei Iterationszahl iter_nber, rufen die Funktion jedes subplot Zeichnung:

fig, axarr = module.graph_function(fig,axarr,iter_nber,some_parameters, some_data) 

3) die Funktion in Frage ist im Grunde wie folgt; jeder Iteration erzeugt ein Paar Nebenhandlungen auf der gleichen Zeile:

def graph_function(fig,axarr,iter_nber,some_parameters, some_data): 

    axarr[iter_nber,1].plot(--some plotting 1--) 
    axarr[iter_nber,2].plot(--some plotting 2--) 

    return fig,axarr 

Dies funktioniert nicht. Ich habe am Ende der Schleife eine leere Figur. Ich habe verschiedene Kombinationen der oben genannten ausprobiert, wie nur axarr im Argument der Funktion verlassen, ohne Erfolg. Offensichtlich verstehe ich die Logik dieser Figur und ihrer Subplots nicht.

Alle Vorschläge sehr geschätzt.

Answer 1

Der Code, den Sie gepostet haben, scheint weitgehend korrekt zu sein. Abgesehen von der Indizierung, wie @hitzg erwähnt hat, sieht nichts, was du machst, schrecklich ungewöhnlich aus.

Es ist jedoch wenig sinnvoll, das Figuren- und Achsen-Array von Ihrer Plotfunktion zurückzugeben. (Wenn Sie Zugriff auf das Figurenobjekt benötigen, können Sie es immer durch ax.figure bekommen.) Es wird nichts geändert, um sie weiterzugeben und zurückzugeben.

Hier ist ein schnelles Beispiel für die Art der Sache, es klingt wie Sie versuchen zu tun. Vielleicht hilft es etwas Verwirrung zu klären?

import numpy as np 
import matplotlib.pyplot as plt 

def main(): 
    nrows = 3 
    fig, axes = plt.subplots(nrows, 2) 

    for row in axes: 
     x = np.random.normal(0, 1, 100).cumsum() 
     y = np.random.normal(0, 0.5, 100).cumsum() 
     plot(row, x, y) 

    plt.show() 

def plot(axrow, x, y): 
    axrow[0].plot(x, color='red') 
    axrow[1].plot(y, color='green') 

main()