R: manière élégante de déterminer les variables numériques dans une trame de données

Question

Voici le code que j'utilise pour trouver des variables numériques dans une trame de données:

Data <- iris 
numericvars <- NULL 
for (Var in names(Data)) { 
    if(class(Data[,Var]) == 'integer' | class(Data[,Var]) == 'numeric') { 
     numericvars <- c(numericvars,Var) 
    } 
} 
numericvars 

Est-il un moyen moins loufoque de le faire?

Answer 1

C'est un assez simple one-liner avec sapply:

sapply(Data, is.numeric) 
# Sepal.Length Sepal.Width Petal.Length Petal.Width  Species 
#   TRUE   TRUE   TRUE   TRUE  FALSE 

# is.numeric should pick up integer columns too 
Data$Species <- as.integer(Data$Species) 
sapply(Data, is.numeric) 
# Sepal.Length Sepal.Width Petal.Length Petal.Width  Species 
#   TRUE   TRUE   TRUE   TRUE   TRUE 

Answer 2

L'utilisation de sapply() ou lapply() semble logique ici:

sapply(iris, function(x) class(x) %in% c("integer","numeric")) 

qui donne:

> sapply(iris, function(x) class(x) %in% c("integer","numeric")) 
Sepal.Length Sepal.Width Petal.Length Petal.Width  Species 
     TRUE   TRUE   TRUE   TRUE  FALSE 

Worth notant que dans votre boucle, vous développez le numericvars vecteur à chaque itération de la boucle; en R, c'est un gros non-non! Il force R à copier et à étendre le vecteur à chaque fois. Allouer un espace de stockage suffisant avant la main et remplir l'objet; ici que cela signifierait la création numericvars comme

numericvars <- character(length = ncol(iris)) 

puis dans la boucle faisant

nams <- names(iris) 
for(i in seq_len(ncol(iris))) { 
    if(class(iris[, i]) == 'integer' | class(iris[, i]) == 'numeric') { 
     numericvars[i] <- nams[i] 
    } 
} 

Un peu plus de travail, mais beaucoup plus efficace, mais vous ne verrez que quand le nombre d'itérations devient plus grand.

Answer 3

Ceci est un peu plus serré:

R> sapply(colnames(iris), function(x) inherits(iris[,x], c("numeric","integer"))) 
Sepal.Length Sepal.Width Petal.Length Petal.Width  Species 
     TRUE   TRUE   TRUE   TRUE  FALSE 
R> 

Answer 4

Il y a aussi colwise(), numcolwise() et catcolwise() dans plyr. colwise() transforme une fonction qui opère sur un vecteur en une fonction qui fonctionne par colonne sur une trame de données. numcolwise et catcolwise fournissent des versions qui fonctionnent uniquement sur des variables numériques et discrètes respectivement.

library(plyr) 
colwise(is.numeric)(Data) 

> colwise(is.numeric)(Data) 
    Sepal.Length Sepal.Width Petal.Length Petal.Width Species 
1   TRUE  TRUE   TRUE  TRUE FALSE