Распространить строку на несколько столбцов в R

Я пытаюсь выполнить горячее кодирование приведенного ниже символьного кадра данных в 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". Пакет распространял один столбец, а не несколько столбцов, как это было необходимо в этом случае.


r one-hot-encoding
person Vipin    schedule 16.08.2017    source источник
comment
test %>% tibble::rownames_to_column() %>% tidyr::separate_rows(V1) %>% table() доводит вас почти до конца и, возможно, проще, чем ответы здесь.   -  person Axeman    schedule 16.08.2017

Ответы (2)


arrow_upward
1
arrow_downward

Пользовательская функция для распределения значений уникальных строк по столбцам:

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
person Patrik_P    schedule 16.08.2017

arrow_upward
0
arrow_downward

Вот метод с использованием трубы:

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
person Deena    schedule 16.08.2017