【R】データフレームにcharacterやfactorの列が含まれているときに、その列をすべてダミー変数化する関数
タイトル通りの関数を作りましたのでここに置いておきます。
データフレームに文字列が入ってて分析を回せない、、、その列だけ取り出してダミー化してまた結合してって面倒くせえ、、、ってときにお使いください。
{dummies}パッケージを使用していますのでご注意ください。
is.dropは、多重共線性等を気にしたときにTRUEとしてください。1列抜かれます。
convOnlyCharacters2Dummies <- function(data, is.drop = FALSE){ library(dummies) convDummies <- function(data, is.drop){ N <- ncol(data) row_names <- names(data) names_list <- c() new_data <- rep(NA, nrow(data)) for(n in 1:N){ unique_value <- sort(unique(data[,n])) dummied_data <- dummy(data[,n]) if(is.drop == TRUE){ new_data <- cbind(new_data, dummied_data[,-ncol(dummied_data)]) names_list <- c(names_list, paste(row_names[n], unique_value, sep = ".")[-ncol(dummied_data)]) } else { new_data <- cbind(new_data, dummied_data) names_list <- c(names_list, paste(row_names[n], unique_value, sep = ".")) } } new_data <- as.data.frame(new_data) names(new_data) <- c("temp", names_list) return(new_data[,-1]) } varnames <- names(data) todummiesnames <- c() for(name in varnames){ if(class(data[[name]]) == "character" || class(data[[name]]) == "factor"){ todummiesnames <- c(todummiesnames, name) } } dummies_df <- convDummies(data[, todummiesnames], is.drop) df <- cbind(data[, which(!(varnames %in% todummiesnames))], dummies_df) return(df) }