Абстрактный узор Подменю

Question

Я дал сценарий, чтобы расширить и был я вижу, эта картина очень много:

newdf <- df[ which(df$someCol %in% c(values...)), ] 

После ломать голову над этим некоторое время я понял, что это подмножества набора данных, основанный на наличие значений внутри столбца. Чтобы сделать код более удобным для чтения, я хотел отвлечь его на функцию. Например:

subset_column.vals <- function(x, col, selectors) { 
    return(x[which(x$col %in% selectors),]) 
} 

Но функция всегда возвращает фрейм данных с 0 строками. Даже когда выражение с теми же входами вне функции возвращает ожидаемый результат. Кто-нибудь знает, что происходит не так?

Answer 1

Есть уже функции, которые делают это достаточно эффективно. Например:

• использованием dplyr: dplyr::filter(x,col %in% selectors)
• в базовой г: subset(x,col %in% selectors)

Если мы действительно хотим, чтобы написать еще один, это будет работать:

subset_column_vals <- function(x, col, selectors) { 
    return(x[x[,col] %in% selectors,]) 
} 

subset_column_vals(iris,"Species",c("setosa")) 

Обратите внимание, что мы используем [ вместо $ для подмножества data.frame, потому что $ do es не работают хорошо с цитируемыми аргументами.