Python持有matplotlib图以在显示之前传入几个函数

Question

我正在尝试做的是“初始化”一个图，并通过一个函数传递数据集来绘制到这个图上。然后，当我满意的时候，这个图包含了我想要展示的一切，我就展示它。我怎么能这样做呢？

import numpy as np
from matplotlib import pyplot as plt

def plot_polynomials(solutions, train_x, train_y):
    x = np.arange(-5, 6)
    plt.plot(x, genPoly(x, solutions))
    plt.show()

data = read_coords("data.csv")
data = np.asarray(data)
system,solution = pol_regression(data[:,0], data[:,1], 2)
plot_polynomials(solution, data[:,0], data[:,1])

#I want to do something like this but im not sure what plot_polynomials() should contain
#to be able to "hold onto" the constructed graph before showing.
#I need all polynomials on the same graph
for i in range(11):
    system,solution = pol_regression(data[:,0], data[:,1], i)
    plot_polynomials(solution, data[:,0], data[:,1])

Answer 1

我不太确定你所说的“坚持住”是什么意思，但如果你想把所有的东西都画在一个图上，你应该这样做：

如果要堆叠所有回归线，请执行

1. ：

对于范围(11 )中的i，plt.figure()：system，solution = pol_regression(data:,0，data:,1，i) x= np.arange(-5，6) plt.plot(x，genPoly(x，solution))plt.show()

• 如果您想要绘图，请在一个图形( 11个小图形)上绘制11个图形：

对于范围( 11 )中的i: system，solution = pol_regression(data:,0，data:,1，i) plt.subplot(11，1，i) #11行，1列，第i个图x= np.arange(-5，6) plt.plot(x，genPoly(x，solution)) plt.show()

希望这能有所帮助。