R - Невозможно выполнить поиск в функции lapply - PullRequest
Новая
       29

R - Невозможно выполнить поиск в функции lapply

0 голосов
/ 25 июня 2018

Я начинаю с создания пустого фрейма данных: 

results <- data.frame(ID=numeric(0), StartDate=numeric(0), term_type=character(0), EndDate=numeric(0), stringsAsFactors = FALSE)

У меня есть список уникальных идентификационных номеров: uniqueIds <- c (1234, 4566, 7838) </p>

У меня есть функция getDataForGivenId, которая создает фрейм данных в формате: 

ID, StartDate, term_type, EndDate

Мне нужен код для вызова функции getDataForGivenId для каждого идентификатора и добавления результирующего фрейма данных к пустому фрейму данных results.

Я пытался: 

library(dplyr)

results <- bind_rows(results, (lapply(uniqueIds, getDataForGivenId)))

и 

do.call("rbind", lapply(uniqueIds, getDataForGivenId))

и 

for (Id in uniqueIds) {
    Y <- getDataForGivenId(Id)
    results <- rbind(results, Y) 
}

Каждый раз, когда я просто получаю пустой results фрейм данных.

Обратите внимание, что если я возьму вещи из цикла и просто выполню код: 

Y <- getDataForGivenId(1234)
results <- rbind(results, Y)

Я получаю ожидаемый результат.

Кто-нибудь знает, что я делаю не так?

РЕДАКТИРОВАТЬ - мой полный сценарий ниже. 

library(dplyr)
library(lubridate)

enVariables <- Sys.getenv()
username    <- enVariables[["DB_USERNAME"]]
password    <- enVariables[["DB_PASSWORD"]]

results <- data.frame(ID=numeric(0), StartDate=numeric(0), term_type=character(0), EndDate=numeric(0), stringsAsFactors = FALSE)

getConnection <- function(){
  require(RMySQL)
  username <- username
  password <- password 
  con <- dbConnect(
    MySQL(), user=username, password=password, 
    dbname='database', host='host', port=port
  )
  return(con)
}

queryuniqueIds <- "SELECT DISTINCT(id) FROM table LIMIT 5"
con                <- getConnection()
uniqueIds      <- dbGetQuery(con, queryuniqueIds)
dbDisconnect(con)

getDataForGivenID <- function(idNumber) {
  queryData <- paste0(
    "SELECT ",
    "Id, bill_date, bill_hour ",
    "FROM table ",
    "WHERE id = ", idNumber
  ) 
  con <- getConnection()
  Data <- dbGetQuery(con, queryData)
  dbDisconnect(con)
  X <- Data %>% 
    select(ID, bill_date, bill_hour) %>%
    mutate(
      bill_date_x = ymd_hms(bill_date)
    ) %>% 
    arrange(ID, bill_date, bill_hour)
  hour(X$bill_date_x) <- X$bill_hour
  X <- X %>% 
    mutate(
      lag_x     = lag(bill_date_x, 1),
      lag_diff  = difftime(bill_date_x,lag_x, units = "hours") %>% as.integer(),
      lead_x    = lead(bill_date_x, 1),
      lead_diff = difftime(lead_x, bill_date_x, units = "hours") %>% as.integer()
    )
  Y <- X %>% 
    filter(
      is.na(lag_diff) | 
        is.na(lead_diff) | 
        !(lag_diff == 1 & lead_diff == 1),

      is.na(lag_diff) | 
        is.na(lead_diff) | 
        !(lag_diff == 0 | lead_diff == 0)
    ) %>% 
    mutate(
      term_type = "N",
      term_type = replace(term_type, lead_diff == 1, "S"),
      term_type = replace(term_type, lag_diff  == 1, "E")
    )
  Y <- Y %>%
    select(ID, bill_date_x, term_type) %>% 
    mutate(
      lead_date = lead(bill_date_x, 1)
    )  %>% 
    filter(term_type == "S")
  colnames(Y) <- c("ID", "StartDate", "term_type", "EndDate")
  return(Y)
}

do.call("rbind", lapply(uniqueIds, getDataForGivenID))
View(results)

1 Ответ

0 голосов
/ 25 июня 2018

Я наконец понял свою проблему.

Список uniqueIds имел длину 1. R передавался во всем списке как один раз, в результате чего оператор SQL возвращал данные только для первого идентификатора.

Я изменил 

uniqueIds <- dbGetQuery(con, queryuniqueIds)

на 

uniqueIds <- as.data.frame(dbGetQuery(con, queryuniqueIds))

и 

do.call("rbind", lapply(uniqueIds, getDataForGivenId))

на 

results <- do.call("rbind", lapply(uniqueIds$id, getDataForGivenId))

Вещи теперь работаюткак и ожидалось.Спасибо тем, кто помог.

...