Qfyilyi

I have a data frame with many columns of different types. I would like to replace each column with NA of the corresponding class.

for example:

df = data_frame(x = c(1,2,3), y = c("a", "b", "c"))



df[, 1:2] <- NA

yields a data frame with two logical columns, rather than numeric and character.
I know I can tell R:

df[,1] = as.numeric(NA)

df[,2] = as.character(NA)

But how do I do this collectively in a loop for all columns with all possible types of NA?

You can use this "trick" :

df[1:nrow(df),1] <- NA

df[1:nrow(df),2] <- NA

the [1:nrow(df),] basically tells R to replace all values in the column with NA and in this way the logical NA is coerced to the original type of the column before replacing the other values.

Also, if you have a lot of columns to replace and the data_frame has a lot of rows, I suggest to store the row indexes and reuse them :

rowIdxs <- 1:nrow(df)

df[rowIdxs ,1] <- NA

df[rowIdxs ,2] <- NA

df[rowIdxs ,3] <- NA

...

As cleverly suggested by @RonakShah, you can also use :

df[TRUE, 1] <- NA

df[TRUE, 2] <- NA

...

As pointed out by @Cath both the methods still work when you select more than one column e.g. :

df[TRUE, 1:3] <- NA

# or

df[1:nrow(df), 1:3] <- NA

Another solution that applies to all the columns can be to specify the non-NAs and replace with NA, i.e.

df[!is.na(df)] <- NA

which gives,

# A tibble: 3 x 2

      x    y    

  <dbl> <chr>

1    NA <NA> 

2    NA <NA> 

3    NA <NA> 

Using dplyr::na_if:

library(dplyr)



df %>% 

  mutate(x = na_if(x, x),

         y = na_if(y, y))



# # A tibble: 3 x 2

#       x y    

#   <dbl> <chr>

# 1    NA NA   

# 2    NA NA   

# 3    NA NA   

If we want to mutate only subset of columns to NA, then:

# dataframe with extra column that stay unchanged

df = data_frame(x = c(1,2,3), y = c("a", "b", "c"), z = c(4:6))



df %>% 

  mutate_at(vars(x, y), funs(na_if(.,.)))



# # A tibble: 3 x 3

#       x y         z

#   <dbl> <chr> <int>

# 1    NA NA        4

# 2    NA NA        5

# 3    NA NA        6

Another way to change all columns at once while keeping the variables' classes:

df <- lapply(df, function(x) {type <- class(x); x <- NA; class(x) <- type; x})



df

# A tibble: 3 x 2

#      x y    

#  <dbl> <chr>

#1    NA <NA> 

#2    NA <NA> 

#3    NA <NA> 

As @digEmAll notified in comments, there is another similar but shorter way:

df <- lapply(df, function(x) as(NA,class(x)))

Using bind_cols() from dplyr you can also do:

df <- data_frame(x = c(1,2,3), y = c("a", "b", "c"))

classes <- sapply(df, class)

df[,1:2] <- NA



bind_cols(lapply(colnames(x), function(x){eval(parse(text=paste0("as.", classes[names(classes[x])], "(", df[,x],")")))}))



     V1 V2   

  <dbl> <chr>

1    NA NA   

2    NA NA   

3    NA NA 

Please note that this will change the colnames.