Преобразование кадров данных R с использованием переменных в цикле

Я пытаюсь заменить значения в кадре данных R по столбцу. Я хотел бы зациклить заданный список столбцов фрейма данных и заменить все значения «Да» на 1, а все остальные значения на 0.

Я попытался сделать это с помощью функций transform() и ifelse() примерно так:

# List of selected Columns:
ColumnNames = c("Frigori", "Microond" , "Arca", "Aspira")

# Replace Values in dataframe
for(i in 1:length(ColumnNames)){
dataframe <- transform(dataframe, ColumnNames[i] = ifelse(Columnames[i] == "Yes", 1, 0))
}

Этот фрагмент кода отлично работает с явными именами столбцов вне цикла, но с массивом выдает следующую ошибку:

Error: unexpected '=' in:
"for(i in 1:length(Appliances)){
dataframe <- transform(dataframe, ColumnNames[i] ="

Я не знаю, что здесь не так, но проблема должна быть связана с подстановкой переменных.


person Daniel    schedule 12.01.2010    source источник


Ответы (2)


На самом деле код можно упростить до одной короткой строки без циклов или apply() вообще:

dataframe <- data.frame(a = c("No", "Yes", "No", "No", "Yes"),
                        b = c("Hi", "Hi", "Mom", "Hi", "Mom"),
                        c = c("Yes", "Yes", "Yes", "Yes", "No"))
cols <- c("a","c")
dataframe[,cols] <- as.numeric(dataframe[,cols]=="Yes")
dataframe

  a   b c
1 0  Hi 1
2 1  Hi 1
3 0 Mom 1
4 0  Hi 1
5 1 Mom 0
person Ken Williams    schedule 12.01.2010

Смоделированные данные:

data <- data.frame(matrix(ifelse(runif(40)>.5,"YES",letters[1:26]), 10, 4))

Предположим, вы хотите изменить столбцы X2 и X4.

cols <- c("X2","X4")
data[,cols] <- apply(data[cols],2,function(x) ifelse(x=="YES",1,0))
person George Dontas    schedule 20.01.2010