Replacing values using a list in R

Question

I have a large dataframe, and am trying to count up the scores of many questions. Here is some sample data.

Q1 = c("apple", "banana", "cider", "muffin", "chocolate")
Q2 = c("orange", "kiwi", "calzone", "cupcake", "cake")
ID = c("P1", "P2", "P3", "P4", "P5")

mydf = data.frame(Q1,Q2,ID)

answer_key = c("apple", "kiwi", "pizza", "dessert", "cake")

I've been trying to use ifelse and %in% for the whole dataframe

mydf = ifelse(mydf %in% answer_key, 1,0)

but it doesn't work, and it returns a vector when I need a dataframe. I just want to replace my values without having to do this for each question because there are many:

mydf$Q1 <-ifelse(mydf$Q1 == "apple", 1, 0)
mydf$Q2 <-ifelse(mydf$Q2 == "kiwi", 1, 0)

another one: mydf[1:2] <- +(mydf[1:2] == answer_key)

rawr
– rawr

2021-04-10 22:34:11 +00:00
Commented Apr 10, 2021 at 22:34 — rawr
– rawr, Commented Apr 10, 2021 at 22:34

Ian Campbell · Accepted Answer · 2021-04-10 21:39:18Z

2

Perhaps this is what you're looking for?

library(dplry)
mydf %>%
   mutate(across(Q1:Q2,~ +(. %in% answer_key)))
  Q1 Q2 ID
1  1  0 P1
2  0  1 P2
3  0  0 P3
4  0  0 P4
5  0  1 P5

Or a bit messy with base R:

mydf[,c("Q1","Q2")] <- sapply(mydf[,c("Q1","Q2")],function(x) +(x%in%answer_key))
mydf
  Q1 Q2 ID
1  1  0 P1
2  0  1 P2
3  0  0 P3
4  0  0 P4
5  0  1 P5

answered Apr 10, 2021 at 21:39

Ian Campbell

25.2k14 gold badges43 silver badges62 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

akrun Over a year ago

Or another way mydf[1:2] <- +(dim<-(as.matrix(mydf[1:2]) %in% answer_key, dim(mydf[1:2])))

Anoushiravan R · Accepted Answer · 2021-04-10 21:39:12Z

1

I hope this is what you are looking for:

library(dplyr)

mydf %>%
  mutate(across(Q1:Q2, ~ ifelse(.x %in% answer_key, 1, 0)))

  Q1 Q2 ID
1  1  0 P1
2  0  1 P2
3  0  0 P3
4  0  0 P4
5  0  1 P5

edited Apr 10, 2021 at 21:39

answered Apr 10, 2021 at 21:22

Anoushiravan R

22k3 gold badges22 silver badges44 bronze badges

1 Comment

Anoushiravan R Over a year ago

I edited my code since the previous one would not have been working if you had large data set.

Collectives™ on Stack Overflow

Replacing values using a list in R

2 Answers 2

1 Comment

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related