Mutate with custom function in R does not work

Question

I have a data frame, containing a column called: "Frequency". Frequency has values like "Year", "Week", "Month" etc. Now I want to create a new column based on the Frequency column where Year's new corresponding value will be 1, Month's corresponding value will be 12 and Week corresponding value will be 48. I tried to make a function for this as "getValue" and tried to make a new column applying a mutation (dplyr) on that funcion. But unfortunately I am getting the following warning and all values being transformed by the value of "1".

getValue <- function(input) {

          if (input == 'Year')
          {
            result <- 1
          } 
          else if(input == 'Month')
          {
            result <- 12
          } 
          else if(input == 'Week')
          {
            result <- 48
          } 

          return(result)
        }

      Data =  mutate(gateway, YearlyHit = getValue(gateway$Frequency))
      Data

Here is the warning message I am getting -

Warning message:
In if (input == "Year") { :
  the condition has length > 1 and only the first element will be used

How can I achieve the required result in R?

score 2 · Answer 1 · edited Dec 07 '19 at 01:51

2

The function you are passing to mutate, getValue doesn't understand vector data.

Try:

v_getValue <- Vectorize(getValue)    
Data =  mutate(gateway, YearlyHit = v_getValue(gateway$Frequency))

More detail here

edited Dec 07 '19 at 01:51

Stephen Rauch

1,831
11
23
34

answered Dec 07 '19 at 00:17

Antoni Abella Vendrell

21
2

score 1 · Accepted Answer · answered Feb 01 '19 at 15:22

1

Why don't you solve it in this way?

mutate(gateway, YearlyHit = case_when(Frequency == 'Year' ~ 1,
                                      Frequency == 'Month' ~ 12,
                                      Frequency == 'Week' ~ 48)
)

answered Feb 01 '19 at 15:22

3nomis

541
7
17

score 0 · Answer 3 · answered Mar 04 '22 at 21:40

An alternative approach is using mutating and ifelse:

Data for minimum reproducible example:

df = tibble(
            x = 1:10,
            Frequency = c('Year','Week','Month',
            'Month','Year','Week','Year','Year',
            'Week','Month')
            )
# A tibble: 10 x 2
       x Frequency
   <int> <chr>    
 1     1 Year     
 2     2 Week     
 3     3 Month    
 4     4 Month    
 5     5 Year     
 6     6 Week     
 7     7 Year     
 8     8 Year     
 9     9 Week     
10    10 Month
df %>% mutate(YearlyHit = ifelse(Frequency == 'Year', 1,
                  ifelse(Frequency == 'Month', 12,
                  ifelse(Frequency == 'Week', 48,NA))))

Also you can achieve this with %in% operator:

df %>% mutate(YearlyHit = ifelse(Frequency %in% 'Year', 1,
                  ifelse(Frequency %in% 'Month', 12,
                  ifelse(Frequency %in% 'Week', 48,NA))))

Or maybe this is more readable:

df %>%
  mutate(YearlyHit = 1 * (Frequency == 'Year') +
                   12 * (Frequency == 'Month')+
                   48 * (Frequency == 'Week'))
A tibble: 10 x 3
   x Frequency YearlyHit

<int> <chr>         <dbl>
 1     1 Year              1
 2     2 Week             48
 3     3 Month            12
 4     4 Month            12
 5     5 Year              1
 6     6 Week             48
 7     7 Year              1
 8     8 Year              1
 9     9 Week             48
10    10 Month            12

score 0 · Answer 4 · edited Jan 02 '19 at 14:30

0

My question was solved by the following command:

gateway$newcol <- mapply(getValue, gateway$Frequency)

instead of:

Data =  mutate(gateway, YearlyHit = getValue(gateway$Frequency))

edited Jan 02 '19 at 14:30

wacax

3,500
4
26
48

answered Dec 02 '18 at 10:29

Nuibb

125
2
6

Mutate with custom function in R does not work

4 Answers4

A tibble: 10 x 3