COSIE.data_preprocessing.reconstruct_metacell_to_original

reconstruct_metacell_to_original(adata_metacell, metacell_embedding)[source]

Expand metacell-level embeddings back to individual cells based on the metacell-to-original cell mapping.

Parameters

adata_metacellAnnData

An AnnData object generated by metacell_construction(). It must contain:

  • .uns[‘meta_to_original’]: A list of lists, where each sublist contains the indices of original cells belonging to a given metacell.

  • .uns[‘original_cell_num’]: Total number of original cells before metacell construction.

metacell_embeddingnp.ndarray

An array of shape (n_metacells, d), representing the learned embedding for each metacell.

Returns

original_embeddingnp.ndarray

An array of shape (n_original_cells, d), where each original cell inherits the embedding of its corresponding metacell.