Добавить значения к тому же столбцу в цикле for [R] - PullRequest
0 голосов
/ 13 марта 2019

У меня есть фрейм данных с именем dt -

dt <- data.frame(a_check=c(NA,2,1,NA,NA),
                 b_check=c(NA,1,1,NA,1))

Я пытаюсь создать новый столбец с именем error, используя условие ifelse для хранения всех ошибок, добавленных в этот столбец со строкойномер соответствующего NA

образца кода-

for(i in 1:length(colnames(dt))){
  ## NA check for a_check column
  dt$error <- ifelse(is.na(dt[colnames(dt)[i]]),"Missing Value found in a_check on row number - ",NA)
  ## NA check for b_check column
  dt$error <- ifelse(is.na(dt[colnames(dt)[i]]),"Missing Value found in b_check on row number - ",NA)
}

Однако я хочу добавить сообщения об ошибках из a_check & b_check в тот же столбец.

Требуемый вывод-

> dt
  a_check b_check                                           error
1      NA      NA     Missing Value found in a_check on row number - 1 &  Missing Value found in b_check on row number - 1
2       2       1                                            <NA>
3       1       1                                            <NA>
4      NA      NA     Missing Value found in a_check on row number - 4 &  Missing Value found in b_check on row number - 4
5      NA       1     Missing Value found in a_check on row number - 5

** Примечание. Я хочу paste номер строки и хочу добавить сообщения об ошибках в тот же столбец в цикле for.Кроме того, у меня более 500 столбцов, поэтому я использую цикл for.

Ответы [ 2 ]

2 голосов
/ 14 марта 2019

Вы можете попробовать что-то вроде:

library(data.table)
setDT(dt)[, error := if(any(is.na(.SD))) paste(c(.BY$rn, names(dt)[is.na(.SD)]), collapse=" "), 
    by=.(rn=seq_len(dt[,.N]))]

выход:

   a_check b_check                   error
1:      NA      NA       1 a_check b_check
2:       2       1                    <NA>
3:       1       1                    <NA>
4:      NA      NA 4 a_check b_check error
5:      NA       1         5 a_check error
1 голос
/ 14 марта 2019

Если вы зафиксированы, вставьте имена строк в новый столбец, тогда лучше всего использовать цикл for и серию операторов if.Проблема, с которой вы пытаетесь использовать ifelse, заключается в том, что у вас более двух условий: и ошибка, и ошибка, и ошибка b, и ошибки нет.

Решение без использования ifelse

 x<-c()

for(i in 1:nrow(dt)){
  if(is.na(dt[i,"a_check"]) & is.na(dt[i,"b_check"])){
    x<- c(x,paste("Missing Value found in a_check & b_chekc", rownames(dt)[i]))
  }else if(is.na(dt[i,"a_check"])){
    x<- c(x,paste("Missing Value found in a_check", rownames(dt)[i]))
  }else if(is.na(dt[i,"b_check"])){
    x<- c(x,paste("Missing Value found in b_check", rownames(dt)[i]))
  }else{
   x<- c(x,NA)
  }
}

dt$error <- x

Обновление

Как вы указали, есть 500 столбцов, которые не будут работать, поэтому вы можете попробовать это

# Create error message matrix
z<-sapply(colnames(dt), function(i){
ifelse(is.na(x[,i]),paste("Missing Value found in", i, sep =" "), NA)
  })
# Collapse matrix, no error will be an empty string  
error<-apply(z,1,function(i){
  paste(i[!is.na(i)], collapse = " & ")
})

dt$error <- error
...