python - Matplotlib：当同一类中的不同函数调用绘图例程时，将多个图绘制为同一图上的子图？

Question

所以，假设我在 Pendulum 类中有一个函数 plot(self) 调用 plot. 然后我在同一个类 Pendulum 中有另一个函数 plot2(self)，它也调用 plot。我将如何设置 pyplot 以便两个函数将多个图形设置为同一图形上的单独子图？

Answer 1

试试这个：

import numpy as np
import matplotlib.pyplot as plt

class Pendulum:

    def __init__(self):
        self.fig = plt.figure()

    def plot1(self):
        ax = self.fig.add_subplot(211)
        ax.plot([1,2],[3,4])

    def plot2(self):
        ax = self.fig.add_subplot(212)
        ax.plot([1,2],[4,3])

p = Pendulum()
p.plot1()
p.plot2()
plt.show()