synthetic.py 文件源码

def get(self, indices=None, variables=None):

        result = {}

        self.h5_file = h5py.File(self.file_name, 'r')

        if indices is None:
            indices = self.h5_file.keys()
        elif not numpy.iterable(indices):
            indices = [indices]

        if variables is None:
            variables = self.variables
        elif not isinstance(variables, list):
            variables = [variables]

        for cell_id in indices:
            result[cell_id] = {}
            for key in variables:
                result[cell_id][key] = self.h5_file['{c}/{d}'.format(c=cell_id, d=key)][:]

        self.h5_file.close()

        return result