How do I generate a new column using a conditional which evaluates values against a vector?

Question

Let's say I have this column.

dataframe$column<-c(1, 2, 2, 2, 3, 4, 4, 5, 5, 5, 6, 7, 8, 9, NA, NA, 0)

and I got this vector

vector<- c(1, 5, 9)

I need to create a new column in my data frame that would return 0 when the value in any given row is equal to any value of my vector. If it is not equal I would like to return a 0, and if the column originally had a NA, I would like to return a NA as well.

So in my example the new colum should look like this:

[1] 1 0 0 0 0 0 0 1 1 1 0 0 0 1 NA NA 0

I'm pretty sure I could do this with apply or sapply, but I'm not very good with loops.

Thanks in advance.

score 5 · Accepted Answer · answered Mar 04 '22 at 22:56

You can do:

library(tidyverse)

dataframe <- data.frame(column = c(1, 2, 2, 2, 3, 4, 4, 5, 5, 5, 6, 7, 8, 9, NA, NA, 0))

vector<- c(1, 5, 9)

dataframe %>%
  mutate(new_column = case_when(is.na(column) ~ NA_real_,
                                column %in% vector ~ 1,
                                TRUE ~ 0))

   column new_column
1       1          1
2       2          0
3       2          0
4       2          0
5       3          0
6       4          0
7       4          0
8       5          1
9       5          1
10      5          1
11      6          0
12      7          0
13      8          0
14      9          1
15     NA         NA
16     NA         NA
17      0          0

ThomasIsCoding · Answer 2 · 2022-03-04T23:06:25.790

3

Try %in%

> replace(+(x %in% v), is.na(x), NA)
 [1]  1  0  0  0  0  0  0  1  1  1  0  0  0  1 NA NA  0

or outer + rowSums

> +(rowSums(outer(x, v, `==`)) > 0)
 [1]  1  0  0  0  0  0  0  1  1  1  0  0  0  1 NA NA  0

Data

x <- c(1, 2, 2, 2, 3, 4, 4, 5, 5, 5, 6, 7, 8, 9, NA, NA, 0)
v <- c(1, 5, 9)

edited Mar 04 '22 at 23:06

answered Mar 04 '22 at 22:51

ThomasIsCoding

96,636
9
24
81

2

This doesn‘t keep the NA‘s, though. – deschen Mar 04 '22 at 22:58
Thanks, sadly it works patially since when it evaluates a `NA` it returns `FALSE` instead of `NA`, as I`m working with survey data is important for me to make a difference between a missing and a value equal to zero. – esteban Mar 04 '22 at 22:59
@deschen Thanks for pointing it out. Please see my update – ThomasIsCoding Mar 04 '22 at 23:06
1

@esteban Please see my update – ThomasIsCoding Mar 04 '22 at 23:07
Thanks, now it works as intended. – esteban Mar 04 '22 at 23:17
2

`(x %in% vector) * NA^(is.na(x))` – Onyambu Mar 04 '22 at 23:25

How do I generate a new column using a conditional which evaluates values against a vector?

2 Answers2

Related