Raggruppare con split()

Fa parte della sezione Logica e flusso del percorso R di Coddy. Lezione 39 di 64.

split(values, groups) inserisce ogni valore nel gruppo denominato dall'elemento corrispondente di groups. Il risultato è una lista con un elemento per ogni gruppo, denominati in base ai gruppi in ordine alfabetico:

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"]])

Output:

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

Combina split() con sapply() per calcolare un riepilogo per ogni gruppo:

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))

Output:

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

I valori possono essere anche testo, ad esempio per raggruppare i nomi in base alla loro prima lettera:

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")
}

Output:

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

Un gruppo esiste solo quando gli appartiene almeno un valore, quindi controlla con %in% names() prima di leggere un gruppo che potrebbe essere vuoto:

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

Output:

[1] FALSE

Sfida

Facile

Completa category_totals(amounts, categories). Restituisci una riga per categoria in ordine alfabetico, food: 40 (2 items), con l’importo totale e il numero di importi in quella categoria.

Il codice fornito legge gli importi e le categorie come due righe separate da virgole della stessa lunghezza e stampa ogni riga restituita.

Provalo tu

category_totals <- function(amounts, categories) {
  # Write your code here
  character(0)
}

# Supplied input/output code: keep it as it is
input <- suppressWarnings(readLines(file("stdin")))
cat(category_totals(as.numeric(strsplit(input[1], ",")[[1]]), strsplit(input[2], ",")[[1]]), sep = "\n")