r - 在 R 中创建一个由级别名称填充的新列

Question

我有一个包含一列整数的 R 数据框。此列有一组级别:

my_data <- data.frame(x=c(1,2,3,1,2,3,1,2,3))
levels(my_data$x) <- c("group1", "group2", "group3")

如何创建一个新的字符列，其中填充了关卡名称？期望的输出:

> my_data
  x      y
1 1 group1
2 2 group2
3 3 group3
4 1 group1
5 2 group2
6 3 group3
7 1 group1
8 2 group2
9 3 group3

Answer 1

“x”列是添加了附加属性 levels 的数字列(本身不是 factor 列)

 str(my_data$x)
 num [1:9] 1 2 3 1 2 3 1 2 3
 - attr(*, "levels")= chr [1:3] "group1" "group2" "group3"

因此我们使用“x”值(按顺序)作为levels 属性的索引

library(dplyr)
my_data <- my_data %>%
     mutate(y = levels(x)[x])

-输出

my_data
x      y
1 1 group1
2 2 group2
3 3 group3
4 1 group1
5 2 group2
6 3 group3
7 1 group1
8 2 group2
9 3 group3

---

或者在base R

mydata$y <- with(my_data, levels(x)[x])

如果'x'的值不同，使用match

mydata$y <- with(my_data, levels(x)[match(x, unique(x))])