Crea nuove variabili in r con mutate() e case_when()
Spesso potresti voler creare una nuova variabile in un frame di dati in R in base a determinate condizioni. Fortunatamente, questo è facile da fare utilizzando le funzioni mutate() e case_when() dal pacchetto dplyr .
Questo tutorial mostra diversi esempi di utilizzo di queste funzioni con il seguente frame di dati:
#create data frame df <- data.frame(player = c('a', 'b', 'c', 'd', 'e'), position = c('G', 'F', 'F', 'G', 'G'), points = c(12, 15, 19, 22, 32), rebounds = c(5, 7, 7, 12, 11)) #view data frame df player position points rebounds 1 to G 12 5 2 b F 15 7 3 c F 19 7 4 d G 22 12 5th G 32 11
Esempio 1: creare una nuova variabile basata su una variabile esistente
Il codice seguente mostra come creare una nuova variabile chiamata “scorer” in base al valore nella colonna dei punti:
library(dplyr) #define new variable 'scorer' using mutate() and case_when() df %>% mutate (scorer = case_when (points < 15 ~ ' low ', points < 25 ~ ' med ', points < 35 ~ ' high ')) player position points rebounds scorer 1 a G 12 5 low 2 b F 15 7 med 3 c F 19 7 med 4 d G 22 12 med 5th G 32 11 high
Esempio 2: creare una nuova variabile basata su diverse variabili esistenti
Il codice seguente mostra come creare una nuova variabile chiamata “tipo” in base al valore nella colonna giocatore e posizione:
library(dplyr) #define new variable 'type' using mutate() and case_when() df %>% mutate (type = case_when (player == 'a' | player == 'b' ~ ' starter ', player == 'c' | player == 'd' ~ ' backup ', position == 'G' ~ ' reserve ')) player position points rebounds type 1 a G 12 5 starter 2 b F 15 7 starter 3 c F 19 7 backup 4 d G 22 12 backup 5th G 32 11 reserve
Il codice seguente mostra come creare una nuova variabile chiamata “valueAdded” in base al valore delle colonne punti e rimbalzi:
library(dplyr) #define new variable 'valueAdded' using mutate() and case_when() df %>% mutate (valueAdded = case_when (points <= 15 & rebounds <=5 ~ 2, points <=15 & rebounds > 5 ~ 4, points < 25 & rebounds < 8 ~ 6, points < 25 & rebounds > 8 ~ 7, points >=25 ~ 9)) player position points rebounds valueAdded 1 to G 12 5 2 2 b F 15 7 4 3c F 19 7 6 4 d G 22 12 7 5th G 32 11 9
Risorse addizionali
Come rinominare le colonne in R
Come eliminare le colonne in R
Come filtrare le righe in R