def hist_set(self): """ A method to generate and yield (energy, histogram) pairs. Usage: for energy, hist in fits_parser.hist_set(): ... do something ... yield: (float, Histogram): The energy and histogram pairs. This will yield one set for each energy in the fits file. """ print 'Generating hist set' xedges = self._get_edges(self._x_ref, self._x_del) yedges = self._get_edges(self._y_ref, self._y_del) for i, energy in enumerate(self._energies): print '\tHistogram for', energy h = Histogram() h.setup(self._data[i], xedges, yedges) energy = energy[0] yield (energy, h) print 'Done'