Python 如何通过函数传递matplotlib对象；作为轴，轴还是图形？

抱歉，如果这是一个有点冗长的事先，但如果我削减了太多的问题是丢失。我正在尝试在pandas和matplotlib上制作一个模块，它将使我能够制作类似于散点矩阵的剖面图和剖面矩阵。我非常确定我的问题归结到我需要从Profile（）返回什么对象，以便我可以在Profile_Matrix（）中处理轴操作。然后问题是返回什么表单Profile_Matrix（），以便编辑子批次

我的模块（ProfileModule.py）借用了很多东西，看起来像：

import pandas as pd
from pandas import Series, DataFrame
import numpy as np
import matplotlib.pyplot as plt

def Profile(x,y,nbins,xmin,xmax):
    df = DataFrame({'x' : x , 'y' : y})

    binedges = xmin + ((xmax-xmin)/nbins) * np.arange(nbins+1)
    df['bin'] = np.digitize(df['x'],binedges)

    bincenters = xmin + ((xmax-xmin)/nbins)*np.arange(nbins) + ((xmax-xmin)/(2*nbins))
    ProfileFrame = DataFrame({'bincenters' : bincenters, 'N' : df['bin'].value_counts(sort=False)},index=range(1,nbins+1))

    bins = ProfileFrame.index.values
    for bin in bins:
        ProfileFrame.ix[bin,'ymean'] = df.ix[df['bin']==bin,'y'].mean()
        ProfileFrame.ix[bin,'yStandDev'] = df.ix[df['bin']==bin,'y'].std()
        ProfileFrame.ix[bin,'yMeanError'] = ProfileFrame.ix[bin,'yStandDev'] / np.sqrt(ProfileFrame.ix[bin,'N'])

    fig = plt.figure(); 
    ax = ProfilePlot.add_subplot(1, 1, 1)
    plt.errorbar(ProfileFrame['bincenters'], ProfileFrame['ymean'], yerr=ProfileFrame['yMeanError'], xerr=(xmax-xmin)/(2*nbins), fmt=None)

    return ax
    #or should I "return fig"


def Profile_Matrix(frame):

    import pandas.core.common as com
    import pandas.tools.plotting as plots
    from pandas.compat import lrange
    from matplotlib.artist import setp

    range_padding=0.05


    df = frame._get_numeric_data()
    n = df.columns.size

    fig, axes = plots._subplots(nrows=n, ncols=n, squeeze=False)

    # no gaps between subplots
    fig.subplots_adjust(wspace=0, hspace=0)

    mask = com.notnull(df)

    boundaries_list = []
    for a in df.columns:
        values = df[a].values[mask[a].values]
        rmin_, rmax_ = np.min(values), np.max(values)
        rdelta_ext = (rmax_ - rmin_) * range_padding / 2.
        boundaries_list.append((rmin_ - rdelta_ext, rmax_+ rdelta_ext))

    for i, a in zip(lrange(n), df.columns):
        for j, b in zip(lrange(n), df.columns):
            ax = axes[i, j]
            common = (mask[a] & mask[b]).values
            nbins = 100
            (xmin,xmax) = boundaries_list[i]

            ax=Profile(df[b][common],df[a][common],nbins,xmin,xmax)
            #Profile(df[b][common].values,df[a][common].values,nbins,xmin,xmax)

            ax.set_xlabel('')
            ax.set_ylabel('')

            plots._label_axis(ax, kind='x', label=b, position='bottom', rotate=True)
            plots._label_axis(ax, kind='y', label=a, position='left')

            if j!= 0:
                ax.yaxis.set_visible(False)
            if i != n-1:
                ax.xaxis.set_visible(False)

    for ax in axes.flat:
        setp(ax.get_xticklabels(), fontsize=8)
        setp(ax.get_yticklabels(), fontsize=8)

    return axes

这将与以下内容一起运行：

import pandas as pd
from pandas import Series, DataFrame
import numpy as np
import matplotlib.pyplot as plt

import ProfileModule as pm

x = np.random.uniform(0, 100, size=1000)
y = x *x  +  50*x*np.random.randn(1000)
z = x *y  +  50*y*np.random.randn(1000)

nbins = 25
xmax = 100
xmin = 0

ProfilePlot = pm.Profile(x,y,nbins,xmin,xmax)
plt.title("Look this works!")

#This does not work as expected
frame = DataFrame({'z' : z,'x' : x , 'y' : y})
ProfileMatrix = pm.Profile_Matrix(frame)
plt.show()

这将有望产生一个简单的剖面图和一个3x3剖面矩阵，但事实并非如此。我尝试了各种不同的方法来实现这一点，但我认为不值得全部解释

---

我应该提到的是，我正在Windows7上使用Enthough Canopy Express。抱歉发了这么长的帖子，再次感谢您对代码的帮助。这是我使用Python的第一周。

您应该传递

轴

对象，并将函数分解为一次操作一个轴。你很接近，但只是改变

import numpy as np
import matplotlib.pyplot as plt

def _profile(ax, x, y):
    ln, = ax.plot(x, y)
    # return the Artist created
    return ln


def profile_matrix(n, m):
    fig, ax_array = plt.subplots(n, m, sharex=True, sharey=True)
    for ax in np.ravel(ax_array):
        _profile(ax, np.arange(50), np.random.rand(50))

profile_matrix(3, 3)

---

我认为您应该返回整个图形

fig

，您始终可以使用

fig.axes

访问轴，并且您无论如何都需要它，以防您想要调整整个对象或某个对象的大小。在不丢失问题的情况下，这可以减少很多。去掉任何涉及熊猫的内容，使用合成数据。谢谢，但我仍然需要“fig=plt.figure（）；ax=fig.add_子图（1,1,1）”，它不能解决我在Profile_矩阵中遇到的问题。它给出了图中没有分开的9个图。我想这是因为我无法通过ax将Profile（）与轴[I，j]相关联。注释掉的方式也不起作用。您不需要这些行。请编辑您的问题，删除所有pandas呼叫，并将

np.random.rand（50）

作为您的数据传递。这段代码太粗糙，无法清楚地看到发生了什么。