util/stats/barchart.py

41     def __init__(self, default=None, **kwargs):
42         super(BarChart, self).__init__(default, **kwargs)
43         self.inputdata = None
44         self.chartdata = None
45         self.inputerr = None
46         self.charterr = None
48     def gen_colors(self, count):
49         cmap = matplotlib.cm.get_cmap(self.colormap)
63     def set_data(self, data):
65             self.inputdata = None
66             self.chartdata = None
73         self.inputdata = data
78             self.chartdata = array([[data]])
84             self.chartdata = transpose([data], axes=(2,0,1))
90             self.chartdata = transpose(data, axes=(1,2,0))
92     def get_data(self):
93         return self.inputdata
97     def set_err(self, err):
99             self.inputerr = None
100             self.charterr = None
107         self.inputerr = err
110             self.charterr = array([[err]])
113             self.charterr = transpose([err], axes=(2,0,1))
116             self.charterr = transpose(err, axes=(1,2,0))
118     def get_err(self):
119         return self.inputerr
151     def graph(self):
152         if self.chartdata is None:
155         dim = len(shape(self.inputdata))
156         cshape = shape(self.chartdata)
157         if self.charterr is not None and shape(self.charterr) != cshape:
161             colors = self.gen_colors(cshape[2])
165             colors = self.gen_colors(cshape[0])
169             colors = self.gen_colors(cshape[1])
174         self.figure = pylab.figure(figsize=self.chart_size)
178         if self.xsubticks is not None:
179             color = self.figure.get_facecolor()
180             self.metaaxes = self.figure.add_axes(self.figure_size,
182             for tick in self.metaaxes.xaxis.majorTicks:
185             self.metaaxes.set_yticklabels([])
186             self.metaaxes.set_yticks([])
188             size[0] = self.figure_size[0]
189             size[1] = self.figure_size[1] + .12
190             size[2] = self.figure_size[2]
191             size[3] = self.figure_size[3] - .12
192             self.axes = self.figure.add_axes(size)
193             outer_axes = self.metaaxes
194             inner_axes = self.axes
196             self.axes = self.figure.add_axes(self.figure_size)
197             outer_axes = self.axes
198             inner_axes = self.axes
200         bars_in_group = len(self.chartdata)
206         for i,stackdata in enumerate(self.chartdata):
213                 if self.charterr is not None:
214                     yerr = self.charterr[i][j]
215                 bar = self.axes.bar(ind, bardata, width, bottom=bottom,
217                 if self.xsubticks is not None:
218                     self.metaaxes.bar(ind, [0] * len(bardata), width)
223         if self.xlabel is not None:
224             outer_axes.set_xlabel(self.xlabel)
226         if self.ylabel is not None:
227             inner_axes.set_ylabel(self.ylabel)
229         if self.yticks is not None:
230             ymin, ymax = self.axes.get_ylim()
231             nticks = float(len(self.yticks))
234             inner_axes.set_yticklabels(self.yticks)
235         elif self.ylim is not None:
236             inner_axes.set_ylim(self.ylim)
238         if self.xticks is not None:
240             outer_axes.set_xticklabels(self.xticks)
242         if self.xsubticks is not None:
245             xsubticks = list(self.xsubticks) + [ '' ]
249         if self.legend is not None:
258             if self.fig_legend:
259                 self.figure.legend(lbars, self.legend, self.legend_loc,
260                                    prop=FontProperties(size=self.legend_size))
262                 self.axes.legend(lbars, self.legend, self.legend_loc,
263                                  prop=FontProperties(size=self.legend_size))
265         if self.title is not None:
266             self.axes.set_title(self.title)
268     def savefig(self, name):
269         self.figure.savefig(name)
271     def savecsv(self, name):
273         data = array(self.inputdata)
277             #if self.xlabel:
278             #    f.write(', '.join(list(self.xlabel)) + '\n')
281             #if self.xlabel:
282             #    f.write(', '.join([''] + list(self.xlabel)) + '\n')
285                 #if self.ylabel:
286                 #    ylabel = [ self.ylabel[i] ]