0

我有一个数据框，其中包含一个因素，例如：

> var1 <- gl(10, 2, labels=letters[1:10])
> var2 <- c(1:20)
> data <- data.frame(var1=var1,var2=var2)
> data
   var1 var2
1     a    1
2     a    2
3     b    3
4     b    4
5     c    5
6     c    6
7     d    7
...
20    j   20

我正在尝试为特定值生成一个定义为 1 和 0 的二分变量var1。但是，当我输入以下代码时：

> data <- data.frame(var1=var1,var2=var2)
> data$var3 <- c(1[which(var1=="a" | var1=="b" | var1=="c" | var1=="d" | 
var1=="e")], 0[which(var1=="f" | var1=="g" | var1=="h" | var1=="i" | var1=="j")])

我得到以下信息：

> data$var3
 [1]  1 NA NA NA NA NA NA NA NA NA NA NA NA NA NA NA NA NA NA NA

第一项更改为 1，但其余变为 NA。我怎样才能得到我想要的结果？

4

1 回答 1

2

我不确定我什至理解您期望您的代码如何运行，但您似乎只想为来自 var1 的值创建一个 1 的向量，这些值是 a、b、c、d 或 e，以及 0对于其余的。如果是这种情况，那么只需使用%in%，它将创建一个逻辑向量，并将其包装起来as.numeric以将其转换为 1 和 0。

例子：

data$var3 <- as.numeric(data$var1 %in% c("a", "b", "c", "d", "e"))
## Or, shorter:
## data$var3 <- as.numeric(data$var1 %in% letters[1:5])

> head(data, 3)
  var1 var2 var3
1    a    1    1
2    a    2    1
3    b    3    1
> tail(data, 3)
   var1 var2 var3
18    i   18    0
19    j   19    0
20    j   20    0
于 2013-03-23T18:32:25.993 回答