fix(CellDimPlot/FeatureStatPlot): fix when reduction is only on a subset of cells

Author: pwwang

R/celldimplot.R

@@ -265,7 +265,8 @@ CellDimPlot.Seurat <- function(
         graph <- object@graphs[[graph]]
     }
 
-    data <- cbind(Embeddings(object, reduction = reduction), object@meta.data)
+    emb <- Embeddings(object, reduction = reduction)
+    data <- cbind(emb, object@meta.data[rownames(emb), , drop = FALSE])
     if (is.null(group_by)) {
         group_by <- "Identity"
         data[[group_by]] <- Idents(object)

R/featurestatplot.R

@@ -544,7 +544,12 @@ FeatureStatPlot.Seurat <- function(
     if (is.null(reduction)) {
         data <- cbind(object@meta.data, assay_data)
     } else {
-        data <- cbind(Embeddings(object, reduction = reduction), object@meta.data, assay_data)
+        emb <- Embeddings(object, reduction = reduction)
+        data <- cbind(
+            emb,
+            object@meta.data[rownames(emb), , drop = FALSE],
+            assay_data[rownames(emb), , drop = FALSE]
+        )
     }
 
     if (is.null(ident)) {