anndata k折交叉

如何将anndata拆分为k份

import scanpy as sc
import anndata as adclass KSplitAnndata:@staticmethoddef _base_split(data: object, k: int) -> list:adata = data.copy()num = adata.n_obs // kadata_list = []for i in range(k):if num <= adata.n_obs:adata_list.append(sc.pp.subsample(adata, n_obs=num, copy=True))adata = adata[~adata.obs_names.isin(adata_list[i].obs_names)]else:adata_list.append(adata)return adata_list@staticmethoddef k_split(data: object, k: int, batch_key: str = None) -> list:""":param data: anndata object:param k: k fold:param batch_key: split by batch, default is None"""adata = data.copy()if batch_key:adata_batch_list = []adata_list = []batch_tuple = set(adata.obs[batch_key])for batch in batch_tuple:adata_batch = adata[adata.obs[batch_key]==batch]adata_batch_list.append(KSplitAnndata._base_split(adata_batch, k))for i in range(k):adata_list.append(ad.concat([adata_batch[i] for adata_batch in adata_batch_list]))else:adata_list = KSplitAnndata._base_split(adata, k)return adata_list# example
adata_list = KSplitAnndata.k_split(adata, 5, "batch")