Agrupando com split()

Parte da seção Lógica e Fluxo do Journey de R da Coddy. Lição 39 de 64.

split(values, groups) coloca cada valor no grupo nomeado pelo elemento correspondente de groups. O resultado é uma lista com um elemento por grupo, nomeada de acordo com os grupos em ordem crescente:

amounts <- c(10, 20, 30, 40)
category <- c("food", "rent", "food", "fun")
by_cat <- split(amounts, category)
print(names(by_cat))
print(by_cat[["food"]])

Saída:

[1] "food" "fun"  "rent"
[1] 10 30

Combine split() com sapply() para calcular um resumo por grupo:

amounts <- c(10, 20, 30, 40)
category <- c("food", "rent", "food", "fun")
by_cat <- split(amounts, category)
print(sapply(by_cat, sum))
print(sapply(by_cat, length))

Saída:

food  fun rent 
  40   40   20 
food  fun rent 
   2    1    1

Os valores também podem ser textos, por exemplo, para agrupar nomes pela primeira letra:

names <- c("ana", "bo", "al", "bea", "cy")
groups <- split(names, substr(names, 1, 1))
for (letter in names(groups)) {
  cat(letter, ":", paste(groups[[letter]], collapse = ", "), "\n")
}

Saída:

a : ana, al 
b : bo, bea 
c : cy

Um grupo só existe quando pelo menos um valor pertence a ele; portanto, verifique com %in% names() antes de ler um grupo que possa estar vazio:

by_cat <- split(c(5, 7), c("food", "food"))
print("rent" %in% names(by_cat))

Saída:

[1] FALSE

Desafio

Fácil

Complete category_totals(amounts, categories). Retorne uma linha por categoria em ordem alfabética, food: 40 (2 items), com o valor total e a quantidade de valores nessa categoria.

O código fornecido lê os valores e as categorias como duas linhas separadas por vírgulas, de mesmo tamanho, e imprime cada linha retornada.

Experimente você mesmo

category_totals <- function(amounts, categories) {
  # Escreva seu código aqui
  character(0)
}

# Código de entrada/saída fornecido: mantenha-o como está
input <- suppressWarnings(readLines(file("stdin")))
cat(category_totals(as.numeric(strsplit(input[1], ",")[[1]]), strsplit(input[2], ",")[[1]]), sep = "\n")