将行“分箱”到范围中（dplyr/R）

Question

我在尝试将数据集中的行放入“垃圾箱”时遇到了很多困难。例如，假设我有一个带有“var1”和“var2”的数据框“df”：

我想创建一个名为“var3”的新变量，它遵循以下逻辑（R 代码）：

1) if var1 <5 and var2<5 .... then var3 = "a"
2) if var1 between (5,10) and var2 between (5,10) .... then var3 = "b"
3) if var1 > 10 and and var2>10 .... then var3 = "c"

从我发布的上一个问题（If statements with multiple range (R)）中，我尝试了以下逻辑：

library(dplyr)
df %>%
  mutate(var3 = case_when(var1 < 5 & var2 < 5 ~ 'a', 
                          var1 > 5 & var1 < 10 & var2 > 5 & var2 < 10 ~ 'b', 
                          var1 >10 & var2 >10 ~ 'c'))

但是当我检查 df$var3 时，逻辑似乎不正确（即 var3 的某些条目没有任何值。注意：var1 和 va2 的最小可能值为 0）。

有人可以帮帮我吗？

谢谢

更新：

样本数据集：

a <- rnorm(50,10,10)
b <- rnorm(50, 2,8)

var1 = abs(a)
var2 = abs(b)

df = data.frame(var1, var2)

Answer 1

如果你想使用case_when，

library(dplyr)

## data make
set.seed(111)
df = data.frame(var1 = abs(rnorm(50,10,10)), var2 = abs(rnorm(50,2,8)))

## core
df <- df %>%
  mutate(var3 = case_when(var1 < 5 & var2 < 5 ~ 'a', 
                          var1 < 10 & var2 < 10 ~ 'b',
                          TRUE ~ 'c'))
## plot to check
with(df, plot(var1, var2, col = c(2:4)[as.numeric(as.factor(var3))], cex = 0.7))
abline(h = c(5, 10), v = c(5, 10), lty = 2)