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

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

对于脚本来说，通常首选使用面向对象的 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()

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