在R中创建虚拟变量的简单方法

Question

我想知道如何简单地创建虚拟变量.我在假人身上发现了许多类似的问题,但要么是基于一些外部包装或技术.

我有这样的数据:

df <- data.frame(X=rnorm(10,0,1), Y=rnorm(10,0,1))
df$Z <- c(NA, diff(df$X)*diff(df$Y))

ž创建在X和变化在Y.变化的DF即产物中一个新的变量现在我想在DF创建一个虚拟变量d使得如果位:Z <0则d == 1,如果Z> 0,则d == 0.

我试过这样的方式:

df$D <- NA
for(i in 2:10) {
if(df$Z[i] <0 ) {
D[i] ==1
}
if(df$Z[i] >0 ) {
D[i] ==0
}}

这不起作用.我想知道为什么上面的代码不起作用(这很容易做到这一点)以及虚拟变量如何在R中创建而不使用任何外部包而只需要一点点解释.

Answer 1

试试:

df$D<-ifelse(df$Z<0,1,0)
df
            X           Y           Z  D
1  -0.1041896 -1.11731404          NA NA
2  -1.4286604  1.42523717 -3.36753491  1
3   0.3931643 -0.05525477 -2.69719691  1
4  -0.2236541  1.64531526 -1.04894297  1
5   1.1725167  0.80063291 -1.17932089  1
6   0.7571427  0.64072381  0.06642209  0
7   0.4929186  1.25125268 -0.16131645  1
8   0.9715885 -0.54755653 -0.86103574  1
9  -0.2962052 -1.37459521  1.04851438  0
10 -1.4838675 -0.85788632 -0.61367565  1

该ifelse函数有3个参数:要评估的条件df$Z<0,条件为TRUE1时的值,以及条件为FALSE0时的值.该函数是矢量化的,因此在这种情况下它很有效.

Answer 2

我们可以创建一个逻辑向量df$Z < 0,然后通过包装将其强制转换为二进制+.

 df$D <- +(df$Z <0)

或者正如@BenBolker所提到的,规范选项将是

as.numeric(df$Z < 0)

要么

as.integer(df$Z < 0)

基准

set.seed(42)
Z <- rnorm(1e7)
library(microbenchmark)
microbenchmark(akrun= +(Z < 0), etienne = ifelse(Z < 0, 1, 0),
           times= 20L,  unit='relative')
#    Unit: relative
#    expr      min       lq     mean   median      uq      max neval
#   akrun  1.00000  1.00000 1.000000  1.00000 1.00000 1.000000    20
# etienne 12.20975 10.36044 9.926074 10.66976 9.32328 7.830117    20