Question

我想做的事情可以这样写：

import pylab
class GetsDrawn(object):
    def __init__(self):
        self.x=some_function_that_returns_an_array()
        self.y=some_other_function_that_returns_an_array()

    # verison 1: pass in figure/subplot arguments
    def draw(self, fig_num, subplot_args ):
        pylab.figure(fig_num)
        pylab.subplot( *subplot_args )
        pylab.scatter( self.x, self.y)

即。我可以通过图号和子图配置告诉对象“在哪里”绘制自己。

我怀疑传递pylab对象的版本会更灵活长期运行，但不知道要为函数提供什么类型的对象。

Answer 1

我会初始化__init__中的所有轴。将它们保存在列表中，例如self.ax。然后在draw方法中，您可以将绘图命令直接发送到所需的轴对象：

import matplotlib.pyplot as plt

class GetsDrawn(object):
    def __init__(self):
        self.x=some_function_that_returns_an_array()
        self.y=some_other_function_that_returns_an_array()
        self.ax = []
        for i in range(num_figures):
            fig = plt.figure(i)
            self.ax.append(plt.subplot(1, 1, 1))

    # verison 1: pass in figure/subplot arguments
    def draw(self, fig_num, subplot_args ):
        ax = self.ax[fig_num]
        ax.subplot( *subplot_args )
        ax.scatter( self.x, self.y)

顺便说一下，pylab可以用于互动会话，但pyplot是recommend for scripts。

Answer 2

对于脚本，通常首选使用面向对象的api。

例如，您可以让您的函数接收数字：

def draw(fig, sub_plot_args,x,y):
    ax = fig.subplot(*sub_plot_args)
    ax.scatter(x,y)

如果您的函数实际上只在一个轴上绘制，您甚至可以将其作为对象传递：

def draw(ax,x,y):

    ax.scatter(x,y)

要创建数字使用：

import matplotlib.pyplot as plt
fig = plt.figure()

并创建一个带有一个子图的图，使用：

fig, ax = plt.subplots()

如果我没弄错的话，最后一个命令只存在于最近的版本中。

为函数指定matplotlib图形和轴的惯用方法是什么？

2 个答案: