r 重新分配整数

Posted 2021-05-23

redistribute_int <- function(vector, new_n) {
  
  # Start rounding down
  ratio <- new_n / sum(vector)
  vector_2 <- (ratio * vector) 
  vector_3 <- vector_2 %>% floor()
  
  # Then redistribute the leftovers over the numbers
  # closer to the next integer
  diff_n <- sum(vector_2) - sum(vector_3)
  diff_vector <- (vector_2 - vector_3)
  higuest_index <- 
    sort(diff_vector, index.return = TRUE, decreasing = TRUE)[[2]][1:diff_n]
  vector_3[higuest_index] <- vector_3[higuest_index] + 1
  
  return(vector_3)
}