r - 使用 dplyr 在数据帧的选择列中重新编码值

Question

我有一个大型数据集，就像下一个示例中的数据集一样。标题中带有 As 的列的代码从 1 到 4，带有 B 的列的代码从 1 到 3。

library(dplyr)

d <- data.frame(
     ID = 1:10,
     A = sample(x = 1:4, size = 10, replace = T),
     AA = sample(x = 1:4, size = 10, replace = T),
     B = sample(x = 1:3, size = 10, replace = T),
     BB = sample(x = 1:3, size = 10, replace = T)
) 

是否有一种巧妙的方法可以在 dplyr 中使用管道将标题中带有 As 的列和带有 Bs 的列中的值重新编码为以下字符串？

• As - 分别从 1、2、3、4 到绿色、黄色、橙色、红色
• Bs - 分别从 1、2、3 到绿色、黄色、红色

这是真实数据集的简化且更友好的版本。

标签： rdplyrpipeline

Answer 1

通过使用 dplyr 中的 mutate_at，可以完成从数字代码到字符串的重新编码。有必要首先将我们要重新编码的列从数字强制转换为字符，否则会出现错误消息。

library(dplyr)

d <- data.frame(
     ID = 1:10,
     A  = sample(x = 1:4, size = 10, replace = T),
     AA = sample(x = 1:4, size = 10, replace = T),
     B  = sample(x = 1:3, size = 10, replace = T),
     BB = sample(x = 1:3, size = 10, replace = T))

d_recoded <- d %>% mutate_at(vars(-contains("ID")), funs(as.character)) %>%
             mutate_at(vars(contains("A"), -contains("ID")), funs(case_when(. == 1 ~ "Green", . == 2 ~ "Yellow", . == 3 ~ "Orange", . == 4 ~ "Red"))) %>%
             mutate_at(vars(contains("B"), -contains("ID")), funs(case_when(. == 1 ~ "Green", . == 2 ~ "Yellow", . == 3 ~ "Red")))