如何通过自定义函数传入数据框与向量名引用对应向量？

嘿，这个坑我之前踩过好多次！你写的函数返回NULL的核心原因是：data.set$column里的column会被R当作字面意义上的列名——也就是说，它会去data.set里找一个名叫column的列，而不是你传入的参数a对应的那列。

下面给你两种简单好用的修改方案：

方案一：用双括号[[ ]]替代$ #

[[ ]]是R里提取列表/数据框元素的标准方式，它支持用变量作为索引，完美适配你的需求：

show_vector <- function(data.set, column) {
  data.set[[column]]
}

现在调用show_vector(df, a)就能得到你想要的结果了：

> show_vector(df, a)
[1] 1 2 3

而且这个方案还兼容字符串形式的列名，比如你传入show_vector(df, "a")也能正常返回结果。

方案二：tidyverse风格的写法（可选） #

如果你平时常用tidyverse系列包，也可以用rlang包的enquo()函数来处理符号参数，代码如下：

library(rlang)
show_vector <- function(data.set, column) {
  data.set[[enquo(column)]]
}

这个写法在更复杂的tidyverse函数场景里会更灵活，但对于你的基础需求来说，方案一已经完全够用啦。

内容的提问来源于stack exchange，提问作者JimmyTheRobot