Создание команды 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» будут переменной, подлежащей анализу на «год», но я не беспокоюсь об этом коде прямо сейчас.

То, что я смущен, есть; как мне сказать 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» теперь будет содержать столбцы year, complex и tot_male gsg_lek.csv, соответствующие 2006 году