Performing dplyr mutate on subset of columns

by

Am I missing something or would this work as expected:

cols <- paste0("X", c(2,4))
dd %>% mutate(evensum = rowSums(.[cols]), evenmean = rowMeans(.[cols]))
#  id        X1        X2        X3        X4        X5   evensum  evenmean
#1  a 0.6021140 0.3670719 0.6872308 0.5090904 0.4474437 0.8761623 0.4380811
#2  b 0.1950439 0.9888592 0.8314290 0.7066286 0.9646670 1.6954878 0.8477439
#3  c 0.9664587 0.8151934 0.1046694 0.8623137 0.1411871 1.6775071 0.8387535
#4  d 0.6509055 0.2539684 0.6461509 0.8417851 0.7767125 1.0957535 0.5478768

Or are you specifically looking for a custom function to do this?

Not exactly what you are looking for but if you want to do it inside a pipe you could use select explicitly inside mutate like this:

dd %>% mutate(xy = select(., num_range("X", c(2,4))) %>% rowSums)
#  id        X1        X2        X3        X4        X5        xy
#1  a 0.6021140 0.3670719 0.6872308 0.5090904 0.4474437 0.8761623
#2  b 0.1950439 0.9888592 0.8314290 0.7066286 0.9646670 1.6954878
#3  c 0.9664587 0.8151934 0.1046694 0.8623137 0.1411871 1.6775071
#4  d 0.6509055 0.2539684 0.6461509 0.8417851 0.7767125 1.0957535

However, it is a bit more complicated if you want to apply several functions. You could use a helper function along the lines of (..not thoroughly tested.. ):

f <- function(x, ...) {
  n <- nrow(x)
  x <- lapply(list(...), function(y) if (length(y) == 1L) rep(y, n) else y)
  matrix(unlist(x), nrow = n, byrow = FALSE)
}

And then apply it like this:

dd %>% mutate(xy = select(., num_range("X", c(2,4))) %>% f(., rowSums(.), max(.)))
#  id        X1        X2        X3        X4        X5      xy.1      xy.2
#1  a 0.6021140 0.3670719 0.6872308 0.5090904 0.4474437 0.8761623 0.9888592
#2  b 0.1950439 0.9888592 0.8314290 0.7066286 0.9646670 1.6954878 0.9888592
#3  c 0.9664587 0.8151934 0.1046694 0.8623137 0.1411871 1.6775071 0.9888592
#4  d 0.6509055 0.2539684 0.6461509 0.8417851 0.7767125 1.0957535 0.9888592

More Related Contents:

  1. Gather multiple sets of columns
  2. How to get summary statistics by group
  3. Is there a dplyr equivalent to data.table::rleid?
  4. Replace NA with previous or next value, by group, using dplyr
  5. Filter rows which contain a certain string
  6. Group by multiple columns in dplyr, using string vector input
  7. Calculate group mean, sum, or other summary stats. and assign column to original data
  8. Rolling mean (moving average) by group/id with dplyr
  9. Programming with dplyr using string as input
  10. Count number of rows by group using dplyr
  11. Pass a vector of variable names to arrange() in dplyr
  12. Efficient way to filter one data frame by ranges in another
  13. R – add column that counts sequentially within groups but repeats for duplicates
  14. Categorize numeric variable with mutate
  15. Means multiple columns by multiple groups [duplicate]
  16. How to use map from purrr with dplyr::mutate to create multiple new columns based on column pairs
  17. Conditional cumsum with reset
  18. How to speed up subset by groups
  19. Use rle to group by runs when using dplyr
  20. Number of significant digits in dplyr summarise
  21. Calculate group mean while excluding current observation using dplyr
  22. Emulate split() with dplyr group_by: return a list of data frames
  23. dplyr mutate rowwise max of range of columns
  24. dplyr: lead() and lag() wrong when used with group_by()
  25. Reorder rows using custom order
  26. Spreading a two column data frame with tidyr
  27. R dplyr rowwise mean or min and other methods?
  28. Run a custom function on a data frame in R, by group
  29. Dealing with spaces and “weird” characters in column names with dplyr::rename()
  30. Filtering rows in R unexpectedly removes NAs when using subset or dplyr::filter

Leave a Comment