common.py 文件源码

def read_tissue_smushed(self, folder, verbose=False, tissue=None):
        smusheds = {}

        if tissue is None:
            globber = glob.iglob(os.path.join(folder, 'smushed-*'))
        else:
            globber = glob.iglob(os.path.join(folder, f'smushed-{tissue}*'))

        for filename in globber:
            if verbose:
                print(f'Reading {filename} ...')
            tissue = filename.split('smushed-')[-1].split('.')[0]
            tissue = tissue.split('-')[0]
            df = pd.read_csv(filename, index_col=0)
            df.rename(columns={'0': 0, '1': 1}, inplace=True)
            smusheds[tissue] = df
            assert len(df.columns.difference([0, 1, 'cluster'])) == 0
        return smusheds