創建一個返回列的函數

Question

我需要創建一個函數來返回給函數作爲參數的列。例如

fun = function (parameter) { 
result = get(paste0("dataframe","parameter")) 
return(result) } 

所以，如果我會用fun (columnnamethatexists)它將返回dataframe$columnnamethatexists

目前，它的作用是返回數據框$參數不存在，不管是什麼，我把作爲輸入。

對不起，這可能是一個愚蠢的複製品，但我找不到它。

謝謝！

Answer 1

dataframe <- data.frame(a = 1:10, b = LETTERS[1:10]) 

fun <- function(val) { 
    dataframe[, val] 
} 

fun("a") 
# [1] 1 2 3 4 5 6 7 8 9 10 
fun("b") 
# [1] A B C D E F G H I J 
# Levels: A B C D E F G H I J 

Answer 2

我認爲df也應該是你函數的一個參數。此解決方案始終以data.frame的形式返回列。

所以：

fun <- function(df, col){ 
    df[col] 
} 
fun(iris, "Sepal.Length") 
    Sepal.Length 
1   5.1 
2   4.9 
3   4.7 
4   4.6 
5   5.0 
6   5.4 
7   4.6 
8   5.0 
... 

你也可以用tidyeval做到這一點，在rlang與dplyr：

library(rlang) 
library(dplyr) 

fun <- function(df, col){ 
    col <- enquo(col) 
    df %>% select(!!col) 
} 

fun(iris, Sepal.Length) 

最佳，

科林