Como digo ao Matplotlib para criar um segundo (novo) enredo e depois mais tarde no antigo?

Eu quero plotar dados, criar uma nova figura e plotar dados2 e, finalmente, voltar ao plot e plotar dados3 originais, mais ou menos assim:

import numpy as np
import matplotlib as plt

x = arange(5)
y = np.exp(5)
plt.figure()
plt.plot(x, y)

z = np.sin(x)
plt.figure()
plt.plot(x, z)

w = np.cos(x)
plt.figure("""first figure""") # Here's the part I need
plt.plot(x, w)

Para sua informação, como posso dizer ao matplotlib que terminei com uma trama? faz algo semelhante, mas não exatamente! Não me permite ter acesso a esse enredo original.

Se você se encontra fazendo coisas assim regularmente, pode valer a pena investigar a interface orientada a objetos para o matplotlib. No seu caso:

import matplotlib.pyplot as plt
import numpy as np

x = np.arange(5)
y = np.exp(x)
fig1, ax1 = plt.subplots()
ax1.plot(x, y)
ax1.set_title("Axis 1 title")
ax1.set_xlabel("X-label for axis 1")

z = np.sin(x)
fig2, (ax2, ax3) = plt.subplots(nrows=2, ncols=1) # two axes on figure
ax2.plot(x, z)
ax3.plot(x, -z)

w = np.cos(x)
ax1.plot(x, w) # can continue plotting on the first axis

É um pouco mais detalhado, mas é muito mais claro e fácil de acompanhar, especialmente com várias figuras, cada uma com várias subparcelas.

Quando você ligar figure, basta numerar o gráfico.

x = arange(5)
y = np.exp(5)
plt.figure(0)
plt.plot(x, y)

z = np.sin(x)
plt.figure(1)
plt.plot(x, z)

w = np.cos(x)
plt.figure(0) # Here's the part I need
plt.plot(x, w)

Editar: observe que você pode numerar as plotagens da maneira que desejar (aqui, a partir de 0), mas se você não fornecer um número para a figura quando criar uma nova, a numeração automática começará em 1("Estilo Matlab" de acordo com para os documentos).

No entanto, a numeração começa em 1, portanto:

x = arange(5)
y = np.exp(5)
plt.figure(1)
plt.plot(x, y)

z = np.sin(x)
plt.figure(2)
plt.plot(x, z)

w = np.cos(x)
plt.figure(1) # Here's the part I need, but numbering starts at 1!
plt.plot(x, w)

Além disso, se você tiver vários eixos em uma figura, como subtramas, use o axes(h)comando onde hestá a alça do objeto de eixos desejado para focar nesses eixos.

(ainda não tem privilégios para comentários, desculpe-me pela nova resposta!)

Uma maneira que encontrei após algumas dificuldades é criar uma função que obtém a matriz data_plot, o nome do arquivo e a ordem como parâmetro para criar gráficos de caixas a partir dos dados fornecidos na figura ordenada (ordens diferentes = figuras diferentes) e salvá-la no nome do arquivo.

def plotFigure(data_plot,file_name,order):
    fig = plt.figure(order, figsize=(9, 6))
    ax = fig.add_subplot(111)
    bp = ax.boxplot(data_plot)
    fig.savefig(file_name, bbox_inches='tight')
    plt.close()