rdplyrmagrittr

how to transform a string into a factor and sets contrasts using dplyr/magrittr piping


i have a rather specific question: how can I make a string into a factor and set its contrasts within a pipe ?

Let's say that I have a tibble like the following

tib <- data_frame (a = rep(c("a","b","c"),3, each = T), val = rnorm(9))

Now, I could use then two separate lines

tib$a <- factor(tib$a)
contrasts(tib$a) <- contr.sum(3)

But what If I wanted to perform the same actions within a pipe from dplyr ?


Solution

  • Ok, this was a fun puzzle, since I never used do() before, but this works for me:

    tib <- data.frame (a = rep(c("a","b","c"),3, each = T), val = rnorm(9)) 
    
    tib = tib %>% mutate(a = factor(a)) %>% do({function(X) {contrasts(X$a) <- contr.sum(3); return(X)}}(.))
    
    contrasts(tib$a)
    

    result:

      [,1] [,2]
    a    1    0
    b    0    1
    c   -1   -1
    

    Hope this helps!

    EDIT: Comment request for explanation, see below:

    This was also new for me. As I understand it, within the do() call, it says

    {func}(.)
    

    this means that a function should be called, with argument ., which is the dataframe in a do call. within func, we then specify the function as

    function(X) {operation to perform on X}
    

    so adding this together:

    {function(X) {operation to perform on X}}(.)
    

    Which means . is used as argument in function X, so it basically becomes 'operation to perform on .'