увеличение ширины фигуры в matplotlib в соответствии с количеством значений x - PullRequest
Новая
       16

увеличение ширины фигуры в matplotlib в соответствии с количеством значений x

4 голосов
/ 18 августа 2011

Я пытался нарисовать barchart, используя matplotlib. Количество отображаемых элементов может варьироваться. Я не могу установить figure.set_size_inches(w,h) или set_figwidth(w) с константами (например, 6., 8. Или 8., 12 и т. Д.), Так как не могу заранее сказать, каким должно быть значение w или h. Я хочу width of figure to increase as the number of items to be plotted increases .Может кто-нибудь сказать мне, как я могу это сделать? 

import pylab

def create_barchart(map):
    xvaluenames = map.keys()
    xvaluenames.sort()
    yvalues = map.values()
    max_yvalue = get_max_yvalue(yvalues)
    xdata = range(len(xvaluenames))
    ydata = [map[x] for x in xvaluenames]
    splitxdata = [x.split('-',1) for x in xvaluenames]
    xlabels = [x[0] for x in splitxdata]
    figure = pylab.figure()
    ax = figure.add_subplot(1,1,1)
    figsize = figure.get_size_inches()
    print 'figure size1=',figsize,'width=',figsize[0],'height=',figsize[1]
    barwidth = .25
    ystep =  max_yvalue/5
    pylab.grid(True)
    if xdata and ydata:
        ax.bar(xdata, ydata, width=barwidth,align='center',color='orange')
        ax.set_xlabel('xvalues',color='green')
        ax.set_ylabel('yvalues',color='green')
        ax.set_xticks(xdata)
        ax.set_xlim([min(xdata) - 0.5, max(xdata) + 0.5])
        ax.set_xticklabels(xlabels)
        ax.set_yticks(range(0,max_yvalue+ystep,ystep))
        ax.set_ylim(0,max(ydata)+ystep)
    figure.autofmt_xdate(rotation=30)
    figure.savefig('mybarplot',format="png")
    print 'figure size2=',figure.get_size_inches()
    pylab.show()

def get_max_yvalue(yvals):
    return max(yvals) if yvals else 0

если я попробую с небольшим набором предметов, я получу 

if __name__=='__main__':
    datamap = dict(mark=39,jim=40, simon=20,dan=33)    
    print datamap
    create_barchart(datamap)

plot of small set

но если я использую больший набор 

datamap = dict(mark=39,jim=40, simon=20,dan=33) 
additional_values= dict(jon=34,ray=23,bert=45,kevin=35,ned=31,bran=11,tywin=56,tyrion=30,jaime=36,griffin=25,viserys=25)
datamap.update(additional_values)  
create_barchart(datamap)

plot of a larger set

Это выглядит ужасно, Мне интересно, есть ли способ увеличить ширину фигуры, в соответствии с количеством элементов, которые будут нанесены на график, сохраняя ширину столбцов в обоих случаях одинаковой

1 Ответ

5 голосов
/ 18 августа 2011

Вы можете установить ширину при инициализации фигуры: 

# default scale is 1 in your original case, scales with other cases:
widthscale = len(yvalues)/4 
figsize = (8*widthscale,6) # fig size in inches (width,height)
figure = pylab.figure(figsize = figsize) # set the figsize

Замените строку figure = pylab.figure() на три вышеуказанные строки, и вы получите то, что просите.

...