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

Question

所以，假设我在Pendulum类中有一个调用plot的函数图（self）。然后我在同一类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()