Переменные подмножества в кадре данных на основе типа столбца

Question

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

Мое текущее решение, чтобы получить имена переменных с указанными типами:

varlist <- function(df=NULL, vartypes=NULL) { 
    type_function <- c("is.factor","is.integer","is.numeric","is.character","is.double","is.logical") 
    names(type_function) <- c("factor","integer","numeric","character","double","logical") 
    names(df)[as.logical(sapply(lapply(names(df), function(y) sapply(type_function[names(type_function) %in% vartypes], function(x) do.call(x,list(df[[y]])))),sum))] 
} 

функция varlist работает следующим образом:

1. Для каждого запрашиваемого типа и для каждого столбца в вызове фрейма данных «есть. TYPE "
2. Суммирует тесты для каждой переменной (логически отливается до целого числа автоматически)
3. Результат преобразования в логический вектор
4. имена подмножество в кадре

данные и некоторые данные, чтобы проверить:

df <- read.table(file="http://archive.ics.uci.edu/ml/machine-learning-databases/statlog/german/german.data", sep=" ", header=FALSE, stringsAsFactors=TRUE) 
names(df) <- c('ca_status','duration','credit_history','purpose','credit_amount','savings', 'present_employment_since','installment_rate_income','status_sex','other_debtors','present_residence_since','property','age','other_installment','housing','existing_credits', 'job','liable_maintenance_people','telephone','foreign_worker','gb') 
df$gb <- ifelse(df$gb == 2, FALSE, TRUE) 
df$property <- as.character(df$property) 
varlist(df, c("integer","logical")) 

Я спрашиваю, потому что мой код выглядит очень загадочно и трудно понять (даже для меня, и я имею закончил функцию 10 минут назад).

Answer 1

subset_colclasses <- function(DF, colclasses="numeric") { 
    DF[,sapply(DF, function(vec, test) class(vec) %in% test, test=colclasses)] 
} 

str(subset_colclasses(df, c("factor", "integer"))) 

Answer 2

Просто сделайте следующее:

df[,sapply(df,is.factor) | sapply(df,is.integer)]