Как мне написать функцию для манипулирования несколькими фреймами одинаково?

Question

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

Вот 2 примера df:

ex1 <- structure(list(info1 = c("Day", "2018.04.03  10:47:33", "2018.04.03  11:20:04", "2018.04.03  11:35:04"), info2 = c("Status_0", "Ok", "Ok", "Ok"
), X = c(200L, 1L, 2L, 3L), X.1 = c(202.5, 1, 2, 3), X.2 = c(205L, 
1L, 2L, 3L), X.3 = c(207.5, 1, 2, 3), X.4 = c(210L, 1L, 2L, 3L
), X.5 = c(212.5, 1, 2, 3), X.6 = c(215L, 1L, 2L, 3L)), class = "data.frame", row.names = c(NA, -4L))

ex2 <- structure(list(info1 = c("Day", "2018.04.10  12:47:33", "2018.04.10  13:20:04", "2018.04.10  13:35:04"), info2 = c("Status_0", "Ok", "Ok", "Ok"
), X = c(200L, 1L, 2L, 3L), X.1 = c(202.5, 1, 2, 3), X.2 = c(205L, 
1L, 2L, 3L), X.3 = c(207.5, 1, 2, 3), X.4 = c(210L, 1L, 2L, 3L
), X.5 = c(212.5, 1, 2, 3), X.6 = c(215L, 1L, 2L, 3L)), class = "data.frame", row.names = c(NA, -4L))

Вот функциональный код для манипулирования 'ex1'

library(tidyverse)
library(lubridate)

  colnames(ex1) <- ex1[1,]                            
  ex1 <- ex1 %>% 
    slice(-1) %>%                                                               
    rename(Date.Time = "Date/Time") %>% 
    mutate(timestamp = parse_date_time(Date.Time, "%Y.%m.%d %H:%M:%S")) %>%     
    select(timestamp, Date.Time, everything()) %>% select(-Date.Time) %>%       
    select(-c(Status_0:"202.5", "212.5":"215"))
  colnames(ex1)[-1] <- paste("raw", colnames(ex1)[-1], sep = "_")

Вторичный вопрос: допустим, я хотел изменить функцию, чтобы она принимала df, но также и тип (то есть, raw или comp), и ввод функции был бы tidydatafunc (df, type). Если я введу type = comp, это изменит последнюю строку кода, где у меня есть «raw» на «comp». Как я мог изменить функцию, чтобы приспособить это?

Любая помощь очень ценится. Я уверен, что это основной материал для большинства из вас!

Answer 1

Оберните ваш скрипт в функцию и укажите параметры.

my_fun <- function(df, type = 'comp') {
  # basic input validation is extremely useful
  stopifnot(is.data.frame(df))
  stopifnot(is.character(type))

  colnames(df) <- df[1,]                            
  ex1 <- df %>% 
  slice(-1) %>%                                                               
  rename(Date.Time = "Date/Time") %>% 
  mutate(timestamp = parse_date_time(Date.Time, "%Y.%m.%d %H:%M:%S")) %>%     
  select(timestamp, Date.Time, everything()) %>% select(-Date.Time) %>%       
  select(-c(Status_0:"202.5", "212.5":"215"))
  # pass the character type
  colnames(df)[-1] <- paste(type, colnames(df)[-1], sep = "_")
  return(df)
}

Тогда вы можете использовать его.

my_fun(ex1, "comp") # view
new_ex1 <- my_fun(ex1, "comp") # save to variable new_ex1