在自定义函数中创建 read() 命令

Question

我仍然是 R 世界的新手，在一个非常加速的课程中，指导有限/没有。我的任务是构建一个读取特定 .csv 的自定义函数，并取出一些特定的列进行分析。有人可以提供一些建议吗？我得到的“示例代码”如下所示：

AnnualLekSurvey=function(data.in,stat.year){
d1=subset(data.in,year==stat.year)
d2=d1[c("year","complex","tot_male")]
      attach(d2)}

所以当它完成并运行它时，我应该可以说：

AnnualLekSurvey(gsg_lek,2006)

其中“gsg_lek”是我要导入的文件的名称，2006 是我想要子集的“年份”列中的值。 “complex”和“tot_male”将是“year”要分析的变量，但我现在不担心那个代码。

我感到困惑的是；我如何告诉 R gsg_lek 是一个 .csv 文件，并告诉它在我运行自定义函数时在正确的目录中查找它？

我在这里看到了另一个模糊相似的例子，他们不得不使用 if() 和 paste() 命令来构建文件名的字符串 - 这似乎太多武断的工作，除非我只是在懒惰……

任何帮助将不胜感激。

Answer 1

你可以做一个这样的函数：

AnnualLekSurvey <- function(csvFile, stat.year)
{
  d1 <- read.csv(paste("C:/",csvFile,".csv", sep=""),header=T, sep=",")  
  d2 <- subset(d1, year==stat.year)
  d2 <- d2[, c("year","complex","tot_male")]
  return(d2)
}

函数中的参数“csvFile”是您的 csv 文件的基本名称。在此特定示例中，它必须位于您的 C:/ 文件夹中。如果您的文件位于其他文件夹中，则必须将函数中的“C:/”更改为 csv 文件所在的文件夹。

运行函数：

data <- AnnualLekSurvey("gsg_lek", "2006")

请注意，参数必须在引号内。 'data' 现在将包含 gsg_lek.csv 的 year、complex 和 tot_male 列，对应于 2006 年