将字符串传播到R中的多列[重复]答案

【问题标题】：Spread a string to multiple columns in R [duplicate]将字符串传播到R中的多列[重复]
【发布时间】：2018-01-24 15:45:30
【问题描述】：

我正在尝试在 R 中对以下字符数据帧进行一次热编码。

x1 <- c('')
x2 <- c('A1,A2')
x3 <- c('A2,A3,A4')
test <- as.data.frame(rbind(x1,x2,x3))

我正在尝试将数据转换为格式：

x1 <- c(0,0,0,0)
x2 <- c(1,1,0,0)
x3 <- c(0,1,1,1)
result <- as.data.frame(rbind(x1,x2,x3))
names(result) = c('A1','A2','A3','A4')

使用的分隔符是逗号，我可以使用逗号分隔：

test$V1 = as.character(test$V1)
split_list = strsplit(test$V1, ",")

这给了我一个无法直接强制转换为数据框的列表列表。有没有更好的方法来做到这一点。我正在尝试“https://www.rdocumentation.org/packages/CatEncoders/versions/0.1.0/topics/OneHotEncoder.fit”。在这种情况下，该包是根据需要展开单个列而不是多个列。

【问题讨论】：

test %>% tibble::rownames_to_column() %>% tidyr::separate_rows(V1) %>% table() 让您快到那里，也许比这里的答案更简单。

标签： r one-hot-encoding

【解决方案1】：

将唯一字符串值分散到列中的自定义函数：

x1 <- c('')
x2 <- c('A1,A2')
x3 <- c('A2,A3,A4')
test <- data.frame(col1=rbind(x1,x2,x3), stringsAsFactors = F) # test$col1 is a character column

cast_variables <- function(df, variable){
  df[df==""] <- "missing" #handling missingness
  x <- as.character(unique(df[[variable]]))
  x <- gsub(" ", "", toString(x)) #so it can split on strings like "A1,A2" and "A1, A2"
  x <- unlist(strsplit(x, ","))
  x <- as.character(x)
  new_columns <- unique(sort(x))[-grep("missing", unique(sort(x)))]
   for (i in seq_along(new_columns)){
    df$temp <- NA
    df$temp <- ifelse(grepl(new_columns[i], df[[variable]]), 1, 0)
    colnames(df)[colnames(df) == "temp"] <- new_columns[i]
  }
  return(df)
}

test <- cast_variables(test, "col1")
print(test)
#       col1 A1 A2 A3 A4
#x1  missing  0  0  0  0
#x2    A1,A2  1  1  0  0
#x3 A2,A3,A4  0  1  1  1

【讨论】：

【解决方案2】：

这是一个使用管道的方法：

library(dplyr)
library(tidyr)
library(reshape2)
library(data.table)

test$V1 %>% 
  strsplit(., ",") %>% 
  setNames(row.names(test)) %>% 
  melt(value.name = 'variable') %>% 
  mutate(dummy = 1) %>% 
  spread(key = variable, value = dummy) %>% 
  list(data.frame(L1 = rownames(test)[!rownames(test) %in% .[['L1']]]), .) %>% 
  rbindlist(., use.names = T, fill = T) %>%
  mutate_all(funs(replace(., is.na(.), 0)))

# L1 A1 A2 A3 A4
# 1 x1  0  0  0  0
# 2 x2  1  1  0  0
# 3 x3  0  1  1  1

【讨论】：