подведение итогов data.table - создание нескольких подмножеств столбцов по дате в R

Между cumsum, by и dcast это почти довольно просто. Самое сложное — это иметь дело с теми месяцами, когда нет данных. Следовательно, это решение не такое краткое, как могло бы быть, но оно делает вещи «способом data.table» и позволяет избежать медленных операций, таких как циклическое перебор строк.

# Just sort the formatting out first
myData[, Date:=as.Date(Date)]
myData[, `:=`(amount = as.numeric(amount),
              year = year(Date),
              month = month(Date))]
bycols <- c('ID', 'year', 'month')

# Summarise all transactions for the same ID in the same month
summary <- myData[, .(amt = sum(amount)), by=bycols]

# Create a skeleton table with all possible combinations of ID, year and month, to fill in any gaps.
skeleton <- myData[, CJ(ID, year, month = 1:12, unique = TRUE)]

# Join the skeleton to the actual data, to recreate the data but with no gaps in
result.long <- summary[skeleton, on=bycols, allow.cartesian=TRUE]
result.long[, amt.cum:=cumsum(fcoalesce(amt, 0)), by=c('ID', 'year')]

# Cast the data into wide format to have one column per month
result.wide <- dcast(result.long, ID + year ~ paste0('amount',month,'month'), value.var='amt.cum')

Примечание. Если у вас нет fcoalesce, обновите пакет data.table.

В каком формате вы хотите? Есть два простых варианта. Вы можете легко получить запрошенный результат в двух разных форматах:

# Prepare the data
ID <- c(rep("A", 5), rep("B", 7), rep("C", 3))
amount <- c(sample(1:1, 15, replace = TRUE))
Date <- c("2016-01-22","2016-07-25", "2016-09-22", "2017-10-22", "2017-01-02", "2016-08-22", "2016-09-22", "2016-10-22", "2017-08-22", "2017-09-22", "2017-10-22", "2018-08-22", "2016-10-22","2017-10-25", "2018-10-22")
myData <- data.frame(ID, amount, Date)
# Add year column
myData$Date <- as.Date(myData$Date, format = "%Y-%m-%d")
myData$year <- format(myData$Date,"%Y")

Обратите внимание, что я изменил amount в целях тестирования. Теперь два решения.

# Format 1
by(myData$amount, list(myData$ID, myData$year), cumsum, simplify = TRUE)
# Format 2
aggregate(myData$amount, list(ID = myData$ID, Date = myData$year), cumsum)

Однако вы можете захотеть, чтобы результатом был новый столбец во фрейме данных? Вы можете решить это:

# Format: New column
myData <- myData[order(myData$year, myData$ID),] # sort by year and ID
myData$cumsum <- rep(0, nrow(myData))
for (r in 1:nrow(myData)) {
  if (r > 1 && myData$year[r-1] == myData$year[r] && myData$ID[r-1] == myData$ID[r])
    myData$cumsum[r] <- myData$cumsum[r-1] + myData$amount[r]
  else
    myData$cumsum[r] <- myData$amount[r]
}

Я не знаю гладкого решения с базовым R. Может быть, у кого-то из "фракции дплр" есть хитрый трюк в рукаве?