Question

我的数据集看起来像

data<-c(0,1,2,3,4,2,3,1,4,3,2,4,0,1,2,0,2,1,2,0,4)
frame<-as.data.frame(data)

我现在想在这个数据框中创建一个新变量。如果列“data”报告的数字为2或更多，我希望它在该行中有“2”，如果有1或0（例如前两个观察），我希望新变量有一个观察结果为“1”。

我正在尝试使用以下代码执行此操作：

frame$twohouses<- if (any(frame$data>=2)) {frame$twohouses=2} else {frame$twohouses=1}

但是，如果我运行这3行脚本，则“twohouses”列中的每个观察都用2编码。但是其中一些应该用1编码。

所以我的问题是：我的if else行或脚本出了什么问题？或者有另一种方法可以做到这一点。

我的问题与此类似： Using ifelse on factor in R

但没有人回答这个问题。

非常感谢提前！

Answer 1

使用ifelse：

frame$twohouses <- ifelse(frame$data>=2, 2, 1)
frame
   data twohouses
1     0         1
2     1         1
3     2         2
4     3         2
5     4         2
...
16    0         1
17    2         2
18    1         1
19    2         2
20    0         1
21    4         2

if和ifelse之间的差异：

if是一个控制流语句，将单个逻辑值作为参数
ifelse是一个向量化函数，以向量作为其所有参数。

if的帮助页面，可通过?"if"访问，也会指向?ifelse

Answer 2

试试这个

frame$twohouses <- ifelse(frame$data>1, 2, 1)
 frame
   data twohouses
1     0         1
2     1         1
3     2         2
4     3         2
5     4         2
6     2         2
7     3         2
8     1         1
9     4         2
10    3         2
11    2         2
12    4         2
13    0         1
14    1         1
15    2         2
16    0         1
17    2         2
18    1         1
19    2         2
20    0         1
21    4         2

在数据框上使用If / Else

2 个答案: