2017-06-27 93 views
1

我的代码存在问题。我有这样的数据帧:for循环中的动态列名称

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 

回答

2

我们可以使用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 
+0

@DominikaJ谢谢你的意见。你也可以检查[这里](https://stackoverflow.com/help/someone-answers) – akrun

0

添加列:

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) 
} 
+0

是的,但我有1000列,在这种情况下需要循环 –

+2

没有必要为这种任务,但几乎不需要''eval' +'parse' - 例如:for(nm in names(data)){data [paste0(nm,“_ NA”)] < - as.integer(is.na(数据[nm]的))}' – thelatemail