5

I need to find a running maximum of a variable by group using R. The variable is sorted by time within group using df[order(df$group, df$time),].

My variable has some NA's but I can deal with it by replacing them with zeros for this computation.

this is how the data frame df looks like:

(df <- structure(list(var = c(5L, 2L, 3L, 4L, 0L, 3L, 6L, 4L, 8L, 4L),
               group = structure(c(1L, 1L, 1L, 1L, 1L, 2L, 2L, 2L, 2L, 2L),
                                 .Label = c("a", "b"), class = "factor"),
               time = c(1L, 2L, 3L, 4L, 5L, 1L, 2L, 3L, 4L, 5L)),
          .Names = c("var", "group","time"),
          class = "data.frame", row.names = c(NA, -10L)))

#    var group time
# 1    5     a    1
# 2    2     a    2
# 3    3     a    3
# 4    4     a    4
# 5    0     a    5
# 6    3     b    1
# 7    6     b    2
# 8    4     b    3
# 9    8     b    4
# 10   4     b    5

And I want a variable curMax as:

var  |  group  |  time  |  curMax
5       a         1         5
2       a         2         5
3       a         3         5
4       a         4         5
0       a         5         5
3       b         1         3
6       b         2         6
4       b         3         6
8       b         4         8
4       b         5         8

Please let me know if you have any idea how to implement it in R.

Jaap
  • 81,064
  • 34
  • 182
  • 193
olga
  • 99
  • 9
  • 1
    Found a possible duplicate of this question. Its answers aren't very good, though: http://stackoverflow.com/q/11675526/1191259 – Frank Dec 03 '15 at 15:19

2 Answers2

8

We can try data.table. Convert the 'data.frame' to 'data.table' (setDT(df1)), grouped by 'group' , we get the cummax of 'var' and assign (:=) it to a new variable ('curMax')

library(data.table)
setDT(df1)[, curMax := cummax(var), by = group]

As commented by @Michael Chirico, if the data is not ordered by 'time', we can do that in the 'i'

setDT(df1)[order(time), curMax:=cummax(var), by = group]

Or with dplyr

library(dplyr)
df1 %>% 
    group_by(group) %>%
    mutate(curMax = cummax(var)) 

If df1 is tbl_sql explicit ordering might be required, using arrange

df1 %>% 
    group_by(group) %>%
    arrange(time, .by_group=TRUE) %>%
    mutate(curMax = cummax(var)) 

or dbplyr::window_order

library(dbplyr)

df1 %>% 
    group_by(group) %>%
    window_order(time) %>%
    mutate(curMax = cummax(var)) 
zero323
  • 322,348
  • 103
  • 959
  • 935
akrun
  • 874,273
  • 37
  • 540
  • 662
  • 3
    Just a robustness check: if it's not necessarily ordered by `time`, you could add `setDT(df1)[order(time), curMax:=cummax(var), by = group]` – MichaelChirico Dec 03 '15 at 15:19
5

you can do it so:

df$curMax <- ave(df$var, df$group, FUN=cummax)
jogo
  • 12,469
  • 11
  • 37
  • 42