6

我以两种不同的格式绘制相同的数据:对数刻度和线性刻度。

基本上我想要完全相同的情节,但比例不同,一个在另一个之上。

我现在拥有的是这样的:

import matplotlib.pyplot as plt

# These are the plot 'settings'
plt.xlabel('Size')
plt.ylabel('Time(s)');
plt.title('Matrix multiplication')

plt.xticks(xl, rotation=30, size='small')
plt.grid(True)

# Settings are ignored when using two subplots

plt.subplot(211)
plt.plot(xl, serial_full, 'r--')
plt.plot(xl, acc, 'bs')
plt.plot(xl, cublas, 'g^')

plt.subplot(212)
plt.yscale('log')
plt.plot(xl, serial_full, 'r--')
plt.plot(xl, acc, 'bs')
plt.plot(xl, cublas, 'g^')

plt.subplot之前的所有“设置”都将被忽略。

我可以让它按照我想要的方式工作,但是我必须在每个子图声明之后复制所有设置。

有没有办法一次配置两个子图?

4

2 回答 2

14

这些plt.*设置通常适用于 matplotlib 的当前绘图;,plt.subplot您正在开始一个新的情节,因此设置不再适用于它。您可以通过浏览Axes与绘图关联的对象来共享标签、刻度等(请参见此处的示例),但恕我直言,这在此处将是矫枉过正。相反,我建议将常见的“样式”放入一个函数中,并在每个绘图中调用它:

def applyPlotStyle():
    plt.xlabel('Size')
    plt.ylabel('Time(s)');
    plt.title('Matrix multiplication')

    plt.xticks(range(100), rotation=30, size='small')
    plt.grid(True)

plt.subplot(211)
applyPlotStyle()
plt.plot(xl, serial_full, 'r--')
plt.plot(xl, acc, 'bs')
plt.plot(xl, cublas, 'g^')

plt.subplot(212)
applyPlotStyle()
plt.yscale('log')
plt.plot(xl, serial_full, 'r--')
plt.plot(xl, acc, 'bs')
plt.plot(xl, cublas, 'g^')

附带说明一下,您可以通过将绘图命令提取到这样的函数中来消除更多重复:

def applyPlotStyle():
    plt.xlabel('Size')
    plt.ylabel('Time(s)');
    plt.title('Matrix multiplication')

    plt.xticks(range(100), rotation=30, size='small')
    plt.grid(True)

def plotSeries():
    applyPlotStyle()
    plt.plot(xl, serial_full, 'r--')
    plt.plot(xl, acc, 'bs')
    plt.plot(xl, cublas, 'g^')

plt.subplot(211)
plotSeries()

plt.subplot(212)
plt.yscale('log')
plotSeries()

另一方面,将标题放在图的顶部(而不是在每个图上)可能就足够了,例如,使用suptitle. xlabel同样,仅出现在第二个图下方可能就足够了:

def applyPlotStyle():
    plt.ylabel('Time(s)');

    plt.xticks(range(100), rotation=30, size='small')
    plt.grid(True)

def plotSeries():
    applyPlotStyle()
    plt.plot(xl, serial_full, 'r--')
    plt.plot(xl, acc, 'bs')
    plt.plot(xl, cublas, 'g^')

plt.suptitle('Matrix multiplication')
plt.subplot(211)
plotSeries()

plt.subplot(212)
plt.yscale('log')
plt.xlabel('Size')
plotSeries()

plt.show()
于 2012-10-18T11:41:03.420 回答
5

汉斯的回答可能是推荐的方法。但是,如果您仍然想将轴属性复制到另一个轴,这是我找到的一种方法:

fig = figure()
ax1 = fig.add_subplot(2,1,1)
ax1.plot([1,2,3],[4,5,6])
title('Test')
xlabel('LabelX')
ylabel('Labely')

ax2 = fig.add_subplot(2,1,2)
ax2.plot([4,5,6],[7,8,9])


for prop in ['title','xlabel','ylabel']:
    setp(ax2,prop,getp(ax1,prop))

show()
fig.show()

在此处输入图像描述

这使您可以设置要设置的属性的白名单,目前我有title,xlabelylabel,但您可以只使用getp(ax1)打印出所有可用属性的列表。

您可以使用以下内容复制所有属性,但我建议不要这样做,因为某些属性设置会弄乱第二个图。我试图使用黑名单来排除一些,但你需要摆弄它才能让它工作:

insp = matplotlib.artist.ArtistInspector(ax1)
props = insp.properties()
for key, value in props.iteritems():
    if key not in ['position','yticklabels','xticklabels','subplotspec']:
        try:
            setp(ax2,key,value)
        except AttributeError:
            pass

except/pass即跳过可获取但不可设置的属性)

于 2012-10-18T19:29:03.270 回答