Question

我有一个看起来像这样的数据框：

codes <- c('TFAA1', 'TFAA2', 'TFAA3', 'TFAA4', 'TFAB1', 'TFAB2', 'TFAB3', 'TFAB4')
scores <- c(4,3,2,2,4,5,1,2)
example <- data.frame(codes, scores)

我想创建一个名为code_group的新列，以TFAA开头的所有内容都称为“ Group1”，以TFAB开头的所有内容都称为“ Group2”。

一直在使用car包中的recode函数和grepl函数，但我失败了。到目前为止，这是我的尝试。...

recode <- (codes, "%in% TFAA='Group1'; %in% TFAB='Group2'")

Answer 1

example$code_group <- ifelse(startsWith(codes, 'TFAA'), 'Group 1', 
                      ifelse(startsWith(codes, 'TFAB'), 'Group 2',
                             NA))

Answer 2

我们可以使用substr提取前四个字符，将其转换为factor，然后将labels指定为我们想要的字符

example$code_group <-  with(example,  as.character(factor(substr(codes, 1, 4), 
              levels = c('TFAA', 'TFAB'), labels = c('Group1', 'Group2'))))

Answer 3

我们可以使用split<-：

example$group <- NA
split(example$group,substr(example$codes,1,4)) <- paste0("Group",1:2)
example
#   codes scores  group
# 1 TFAA1      4 Group1
# 2 TFAA2      3 Group1
# 3 TFAA3      2 Group1
# 4 TFAA4      2 Group1
# 5 TFAB1      4 Group2
# 6 TFAB2      5 Group2
# 7 TFAB3      1 Group2
# 8 TFAB4      2 Group2

或者我们可以将因子用于相同的输出（3个变体）：

example$group <- paste0("Group",factor(substr(example$codes,1,4),,1:2))
example$group <- paste0("Group",as.numeric(factor(substr(example$codes,1,4))))
example$group <- factor(substr(example$codes,1,4),,paste0("Group",1:2))

在最后一种情况下，您将获得一个因子列，在所有其他情况下，您将获得一个字符列。

Answer 4

使用dplyr和stringr，您可以完成它：

library(dplyr)
library(stringr)
example %>% 
  mutate(code_group = case_when(str_detect(codes, "^TFAA") ~ "Group1",
                              str_detect(codes, "^TFAB") ~ "Group2"))

case_when使您可以使用多个if-then情况。 str_detect可以使您很好地检测字符串中要寻找的模式。

根据向量的第一个字符重新编码R中的变量

4 个答案: