dplyr: case_when 涉及很多情况

我有两个数据框：

set.seed(002)
data1 <- data.frame(cbind(
  a1 = sample(letters, 8, replace = TRUE),
  a2 = rpois(8, 10)
), stringsAsFactors = FALSE)

data2 <- data.frame(cbind(
  b1 = paste("area", 1:6, sep = " "),
  b2 = c("e", "s", "o", "y", "d", "v")
), stringsAsFactors = FALSE)

data1
  a1 a2
1  e  9
2  s 10
3  o 12
4  e  9
5  y 16
6  y  9
7  d 11
8  v 13

data2
      b1 b2
1 area 1  e
2 area 2  s
3 area 3  o
4 area 4  y
5 area 5  d
6 area 6  v

我想在 data1 中创建一个名为 a3 的新列，同时将 a1 与 data2 中的信息进行匹配，例如，如果 a1 =“e”则 a3 =“区域 1”，如果 a1 =“d”则 a3 =“区域 5”等等。新的 data1 应如下所示：

  a1 a2     a3
1  e  9 area 1
2  s 10 area 2
3  o 12 area 3
4  e  9 area 1
5  y 16 area 4
6  y  9 area 4
7  d 11 area 5
8  v 13 area 6  

我可以通过这样做来实现这一目标

data1 %>%
  mutate(a3 = case_when(
    a1 == "e" ~ "area 1",
    a1 == "s" ~ "area 2",
    a1 == "o" ~ "area 3",
    a1 == "y" ~ "area 4",
    a1 == "d" ~ "area 5",
    TRUE ~ "area 6"
  ))

问题是我有很多情况，我要在具有不同情况的多个数据帧上重复此操作。

我可以通过写来用基本 r 来做到这一点

data1$a3 <- NA
for(i in 1:nrow(data2)){
  for(j in 1:nrow(data1)){
    if(data1[j,1] == data2[i,2]){
      data1[j,3] <- data2[i,1]
    }
  }
} 

但我很喜欢 dplyr。任何有关如何使用 dplyr 实现此目的的帮助都将受到赞赏。

data1 <- dplyr::left_join(data1, data2, by = c("a1" = "b2"))

data1:-

a1 a2     b1
e  9   area 1
s 10   area 2
o 12   area 3
e  9   area 1
y 16   area 4
y  9   area 4
d 11   area 5
v 13   area 6