diff --git a/anndata/_core/anndata.py b/anndata/_core/anndata.py index 9ba69c724..a00f6951b 100644 --- a/anndata/_core/anndata.py +++ b/anndata/_core/anndata.py @@ -1284,7 +1284,7 @@ def transpose(self) -> AnnData: T = property(transpose) - def to_df(self, layer=None) -> pd.DataFrame: + def to_df(self, layer=None, obs: Optional[Union[str, List[str]]]=None) -> pd.DataFrame: """\ Generate shallow :class:`~pandas.DataFrame`. @@ -1299,6 +1299,8 @@ def to_df(self, layer=None) -> pd.DataFrame: ------ layer : str Key for `.layers`. + obs + Column(s) of `.obs` to include in DataFrame. """ if layer is not None: X = self.layers[layer] @@ -1308,7 +1310,11 @@ def to_df(self, layer=None) -> pd.DataFrame: X = self.X if issparse(X): X = X.toarray() - return pd.DataFrame(X, index=self.obs_names, columns=self.var_names) + + df = pd.DataFrame(X, index=self.obs_names, columns=self.var_names) + if obs is not None: + df = df.merge(self.obs[obs], left_index=True, right_index=True) + return df def _get_X(self, use_raw=False, layer=None): """\