for循環中的動態列名稱

Question

我的代碼存在問題。我有這樣的數據幀：

A <- c(21, 234, NA, 286,NA) 
B <- c(3,NA,NA, 8, 10) 
data <- data.frame(A,B) 
data 

    A B 
1 21 3 
2 234 NA 
3 NA NA 
4 286 8 
5 NA 10 

我想創建的效果是：

A B A_NA B_NA 
1 21 3 0 0 
2 234 NA 0 1 
3 NA NA 1 1 
4 286 8 0 0 
5 NA 10 1 0 

這裏是我的簡單的代碼，但事情並沒有工作..

for(i in c(1:ncol(data))) 
    { 
    data[, ncol(data) + 1] <- ifelse(is.na(data[i]), 1, 0) 
    names(data)[ncol(data)] <- paste0(colnames(data[i]), "_NA") 
    } 

因爲效果是：

A B A A B A A 
1 21 3 0 0 0 0 0 
2 234 NA 0 0 1 0 0 
3 NA NA 1 1 1 0 0 
4 286 8 0 0 0 0 0 
5 NA 10 1 1 0 0 0 

Answer 1

我們可以使用lapply遍歷「數據」的列，檢查元素是否NA（is.na(x)），轉換爲整數（as.integer）和輸出分配給基於條件的新列

data[paste0(names(data), "_NA")] <- lapply(data, function(x) as.integer(is.na(x))) 
data 
# A B A_NA B_NA 
#1 21 3 0 0 
#2 234 NA 0 1 
#3 NA NA 1 1 
#4 286 8 0 0 
#5 NA 10 1 0 

Answer 2

添加列：

data$A_NA<-ifelse(is.na(data$A),1,0) 
data$B_NA<-ifelse(is.na(data$B),1,0) 

此外

遞歸：

for(nm in names(data)) 
    eval(parse(text = paste0("data$",nm,"_NA<-ifelse(is.na(data$",nm,"),1,0)"))) 

個

addition2

另外一個可以使用：

for(nm in names(data)){ 
    assign(paste0(nm,"_NA"), ifelse(is.na(data[nm]),1,0)) 
    tempo<-data.frame(get(paste0(nm,"_NA")));names(tempo)<-paste0(nm,"_NA") 
    data<-cbind(data,tempo) 
}