根据条件创建变量

Question

以iris数据集为例，我试图创建一个新的变量，该变量具有不同的物种分布。

所以，我想创建new.variable，基于随机正态分布，但参数将是不同的每个物种。这是我最近的尝试。我知道我可以拆分数据集，添加变量，然后重新组合，但似乎没有必要。

set.seed(42) 
iris$new.variable <- 0 
iris$new.variable[iris$Species == 'setosa'] <- rnorm(nrow(iris[iris$Species == 'setosa']), 5, 1) 
iris$new.variable[iris$Species == 'versicolor'] <- rnorm(nrow(iris[iris$Species == 'versicolor']), 10, 2) 
iris$new.variable[iris$Species == 'virginica'] <- rnorm(nrow(iris[iris$Species == 'virginica']), 20, 3) 

Answer 1

听起来对我很好。

或者，你可以不喜欢

mns <- c(5, 10, 20) 
sds <- 1:3 
specNum <- as.numeric(iris$Species) 

iris$new.variable <- rnorm(nrow(iris), mns[specNum], sds[specNum])