def __init__(self, fig_label, n_rows, n_plots): GraphicsUnit.__init__(self,n_rows, n_plots) self.label = fig_label # print self.n_cols, self.n_rows, self.label rc('text', usetex=True) rc('font', family='serif') self.figure = plt.figure()
def add_subplot(self, subplot_name): pos = 1 + len(self.subplots ) print subplot_name, self.n_rows, self.n_cols,pos obj = self.figure.add_subplot(self.n_rows, self.n_cols, pos) GraphicsUnit.add_subplot( self, subplot_name, plot_unit = PyplotUnit( plot_object = obj ) )