Update samplev.R

ppehli · web-flow · commit caf289ffecb0 · 2026-04-09T09:49:26.000-04:00
replace column finding with rowsums to save speed
diff --git a/R/samplev.R b/R/samplev.R
@@ -13,6 +13,6 @@ samplev <- function(m_Probs, m = 1) {
   n_samp <- nrow(m_Probs)
   u <- runif(n_samp, min = 0, max = 1)
   v_sum_p <- matrixStats::rowCumsums(m_Probs)
-  v_cat <- lev[max.col(v_sum_p >= u, ties.method = "first")]
+  v_cat <- lev[rowSums(v_sum_p < u) +1]
   return(v_cat)
 }

Original file line number	Diff line number	Diff line change
`@@ -13,6 +13,6 @@ samplev <- function(m_Probs, m = 1) {`
`13`	`13`	`n_samp <- nrow(m_Probs)`
`14`	`14`	`u <- runif(n_samp, min = 0, max = 1)`
`15`	`15`	`v_sum_p <- matrixStats::rowCumsums(m_Probs)`
`16`		`- v_cat <- lev[max.col(v_sum_p >= u, ties.method = "first")]`
	`16`	`+ v_cat <- lev[rowSums(v_sum_p < u) +1]`
`17`	`17`	`return(v_cat)`
`18`	`18`	`}`