不一致的figsize在matplotlib中调整大小

时间:2018-01-30 22:53:44

标签: python python-2.7 matplotlib

我有几个不同的条形图数字,可以生成不同数量的条形图。因此,图形的总宽度和高度会有所不同,但我希望所有条形图的条形尺寸始终相同。

到目前为止我尝试的是按比例调整figsize的数量。这似乎并不一致。

以下是示例代码:

nb_bars_list = [2, 10]

for i, nb_bars in enumerate(nb_bars_list):
    # Resize proportionally to the number of bars
    figsize = [1+nb_bars, 5]
    # Prepare the ticks
    ticks = np.arange(1, 1+nb_bars, 1)
    # Generate random points
    points = [np.random.randn(10) for x in xrange(nb_bars)]
    # Make the plot
    fig, ax = plt.subplots()
    if figsize:
        fig.set_size_inches(figsize[0], figsize[1], forward=True)
    for b in xrange(nb_bars):
        ax.bar(ticks[b], points[b].mean())
    fig.savefig('test%i' % i, bbox_inches='tight')

导致: 2-bars 10-bars

如果我们使用GIMP重叠,我们可以清楚地注意到条宽的差异:

both-bars

无论条数多少,如何确保条的宽度相同?

我正在使用matplotlib 2。

1 个答案:

答案 0 :(得分:5)

要设置图形大小,使图形中不同数量的条形总是具有相同的宽度,则需要考虑图形边距。还需要在所有情况下均等地设置图的xlimits。

import matplotlib.pyplot as plt
import numpy as np

nb_bars_list = [2, 10]
margleft = 0.8 # inch
margright= 0.64 # inch
barwidth = 0.5 # inch


for i, nb_bars in enumerate(nb_bars_list):
    # Resize proportionally to the number of bars
    axwidth = nb_bars*barwidth # inch
    figsize = [margleft+axwidth+margright, 5]
    # Prepare the ticks
    ticks = np.arange(1, 1+nb_bars, 1)
    # Generate random points
    points = [np.random.randn(10) for x in xrange(nb_bars)]
    # Make the plot
    fig, ax = plt.subplots(figsize=figsize)
    fig.subplots_adjust(left=margleft/figsize[0], right=1-margright/figsize[0])

    for b in xrange(nb_bars):
        ax.bar(ticks[b], points[b].mean())
    ax.set_xlim(ticks[0]-0.5,ticks[-1]+0.5)
    #fig.savefig('test%i' % i, bbox_inches='tight')
plt.show()

enter image description here