python - Python：如何使用类来绘制不同的线？

Question

我想使用类方法绘制不同的线。为了绘制一条线。类似于Passing plots out of a class我使用这个类来绘制一条线

from matplotlib.figure import Figure
import matplotlib.pyplot as plt

class Plotter(object):
    def __init__(self, xval=None, yval=None, dim1=None, dim2=None):
        self.xval = xval
        self.yval = yval
        self.dim1 = dim1
        self.dim2 = dim2

    def plotthing(self):
        fig, ax = plt.subplots(figsize=(self.dim1,self.dim2))
        ax.plot(self.xval, self.yval, 'o-')
        return fig

app = Plotter(xval=range(0,10), yval=range(0,10),  dim1=5, dim2=5)
plot = app.plotthing()

但是我想在同一个图中绘制不同的曲线并在类中定义一个函数来做到这一点。

Xval = []
Yval = []
xval=range(0,10)
yval=range(0,10)
Xval.append(xval)
Yval.append(yval)
xval=range(0,10)
yval=np.sin(range(0,10))
Xval.append(xval)
Yval.append(yval)

如何定义要传递给 plotthing 的函数？

class Plotter(object):
    def __init__(self, xval=None, yval=None, dim1=None, dim2=None):
        self.xval = xval
        self.yval = yval
        self.dim1 = dim1
        self.dim2 = dim2

    def function_do_plot(x, y):
        do something  

    def plotthing(self):
        fig, ax = plt.subplots(figsize=(self.dim1,self.dim2))
        for i in range(0, len(self.xval)): 
             x = xval[i]
             y = yval[i]
             fig = function_do_plot(x, y)
        return fig

标签： pythonoopmatplotlib

Answer 1

考虑在不需要单独的方法的情况下遍历输入列表：

class Plotter(object):
    def __init__(self, xval=None, yval=None, dim1=None, dim2=None):
        self.xval = xval    # LIST
        self.yval = yval    # LIST 
        self.dim1 = dim1
        self.dim2 = dim2

    def plotthing(self):
        fig, ax = plt.subplots(figsize=(self.dim1,self.dim2))

        for i, j in zip(self.xval, self.yval):
            ax.plot(i, j, 'o-')

        return fig

# POPULATE LIST OF RANGES
Xval = []; Yval = []

xval = range(0,10)
yval = range(0,10)
Xval.append(xval)
Yval.append(yval)

xval = range(0,10)
yval = np.sin(range(0,10))
Xval.append(xval)
Yval.append(yval)

# PASS IN LIST OF RANGES
app = Plotter(xval=Xval, yval=Yval, dim1=5, dim2=5)
plot = app.plotthing()