根据列值将R中的列拆分为多列

发布于04月20日

我想根据一列的值在df中引入新的COLS.

类别Col需要拆分在‘，’上，并且每个拆分的子串需要转到新引入的适当Col.

最小示例:

df <- data.frame(category=c('1, 2', '1, 3','3', '2, 3'),
                 othercolumn= c("Grealish", "Saka", "Henry", 'Jesus'))

所需的df:

finaldf <- data.frame(category=c('1, 2', '1, 3','3', '2, 3'),
                      category1=c('1', '1',NA, NA),
                      category2=c('2',NA,NA, '2'),
                      category3=c(NA,'3','3', '3'),
                 othercolumn= c("Grealish", "Saka", "Henry", 'Jesus'))

CATEGORY 1、CATEGORY 2和CATEGORY 3 COLS将根据CATEGORY列获取值，否则为NA.

推荐答案

在基数R中，您可以执行以下操作:

使用strsplt拆分数字并将其转换为数字

setName在lapply和cbind中返回到数据帧

nums <- strsplit(df$category, ",") |> 
  type.convert(as.is = T)

cbind(df[1], do.call(rbind, lapply(nums, \(x){
    sq <- seq(max(unlist(nums)))
    ifelse(sq %in% x , sq, NA ) |> 
      setNames(paste0("category", sq))
  })), df[-1])

  category category1 category2 category3 othercolumn
1     1, 2         1         2        NA    Grealish
2     1, 3         1        NA         3        Saka
3        3        NA        NA         3       Henry
4     2, 3        NA         2         3       Jesus