Home R: If my vector grouped by ID contains a certain value, return one value, else return other value

Questions

R: If my vector grouped by ID contains a certain value, return one value, else return other value

June 20, 2022

Each ID variable has multiple rows, and I’d like to create a vector that tells me if any of the runs (rows within that id) contains "orange."

Otherwise, I’d like it to return "apple" if "orange" is not contained on any of the rows for that id.

I’m guessing it’s something like

data_desired <- data %>%
group_by("ID") %>%
mutate(AnyOrange = ...)

but that’s where I’m stuck…sample data and desired outcome below:

library(tidyverse)

data <- tribble(
  ~ID, ~Run, ~Oranges,
  #--/---/---
  "a", 1, "orange",
  "a", 2,  "orange",
  "b", 1, "apple",
  "b", 2, "apple",
  "b", 3, "orange",
  "c", 1, "apple",
  "c", 2, "apple"
)

# Desired Outcome
data <- tribble(
  ~ID, ~Run, ~Oranges, ~AnyOrange,
  #--/---/---/---
  "a", 1, "orange","orange",
  "a", 2,  "orange","orange",
  "b", 1, "apple","orange",
  "b", 2, "apple","orange",
  "b", 3, "orange","orange",
  "c", 1, "apple","apple",
  "c", 2, "apple","apple"
)

>Solution :

data %>%
   group_by(ID) %>%
   mutate(AnyOrange = ifelse(any(Oranges=='orange'), 'orange', Oranges))

# A tibble: 7 x 4
# Groups:   ID [3]
  ID      Run Oranges AnyOrange
  <chr> <dbl> <chr>   <chr>    
1 a         1 orange  orange   
2 a         2 orange  orange   
3 b         1 apple   orange   
4 b         2 apple   orange   
5 b         3 orange  orange   
6 c         1 apple   apple    
7 c         2 apple   apple