dplyr的意外输出（if_else语句）

时间：2018-02-21 16:16:27

标签： r dplyr

我无法弄清楚为什么 if_else 的行为方式，可能是我的代码或数据结构的方式。

下面是我正在进行的数据库的快照，它代表了每周跟进试验的研究参与者的纵向调查。

变量“survey_start”表示研究定义的一年后续行动的开始（我们称之为“survey_year” ）。

我试图在每个调查年度填写每个参与者的所有后续条目，条目“调查”后跟下划线和相应的年份，例如。 survey_2014。

缺少的参赛作品，例如此处所代表的参赛者，在2015年调查开始时无法使用。

我写了两个代码，第一个代码失败而第二个代码正常工作，唯一的区别是我已经颠倒了第二个代码（从2007-2016到2016-2007）填充条目的顺序并删除了if_else 2015年的声明。

请帮助解决这个问题...... 

    trialData <- structure(list(study = c("site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1", "site_1", "site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1", "site_1", "site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1", "site_1", "site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1", "site_1", "site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1", "site_1", "site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1", "site_1", "site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1", "site_1", "site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1", "site_1", "site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1", "site_1", "site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1", "site_1", "site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1", "site_1", "site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1", "site_1", "site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1", "site_1", "site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1", "site_1", "site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1", "site_1", "site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1", "site_1", "site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1", "site_1", "site_1", "site_1", "site_1", "site_1", 
"site_1", "site_1"), studyno = c("child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1", "child_1", "child_1", "child_1", "child_1", 
"child_1", "child_1"), date = structure(c(16078, 16085, 16092, 
16098, 16104, 16115, 16121, 16129, 16135, 16140, 16146, 16156, 
16162, 16168, 16177, 16185, 16191, 16195, 16203, 16210, 16217, 
16225, 16234, 16237, 16246, 16253, 16262, 16269, 16278, 16283, 
16288, 16297, 16304, 16311, 16319, 16326, 16332, 16337, 16346, 
16353, 16360, 16366, 16370, 16381, 16384, 16395, 16399, 16407, 
16415, 16422, 16444, 16452, 16454, 16467, 16474, 16477, 16484, 
16490, 16501, 16508, 16514, 16520, 16529, 16533, 16539, 16550, 
16556, 16564, 16566, 16578, 16582, 16593, 16599, 16604, 16613, 
16620, 16623, 16635, 16636, 16654, 16660, 16666, 16673, 16681, 
16688, 16693, 16702, 16706, 16714, 16721, 16728, 16734, 16745, 
16749, 16757, 16764, 16769, 16778, 16785, 16792, 16805, 16812, 
16819, 16830, 16832, 16839, 16846, 16856, 16862, 16867, 16877, 
16884, 16890, 16898, 16904, 16912, 16917, 16923, 16936, 16938, 
16953, 16960, 16966, 16973, 16980), class = "Date"), year = c(2014L, 
2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 
2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 
2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 
2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 
2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 2014L, 
2014L, 2014L, 2014L, 2014L, 2015L, 2015L, 2015L, 2015L, 2015L, 
2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 
2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 
2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 
2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 
2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 2015L, 
2016L, 2016L, 2016L, 2016L, 2016L, 2016L, 2016L, 2016L, 2016L, 
2016L, 2016L, 2016L, 2016L, 2016L, 2016L, 2016L, 2016L, 2016L, 
2016L, 2016L, 2016L, 2016L, 2016L, 2016L, 2016L), month = c(1L, 
1L, 1L, 1L, 2L, 2L, 2L, 2L, 3L, 3L, 3L, 3L, 4L, 4L, 4L, 4L, 5L, 
5L, 5L, 5L, 5L, 6L, 6L, 6L, 6L, 7L, 7L, 7L, 7L, 8L, 8L, 8L, 8L, 
8L, 9L, 9L, 9L, 9L, 10L, 10L, 10L, 10L, 10L, 11L, 11L, 11L, 11L, 
12L, 12L, 12L, 1L, 1L, 1L, 2L, 2L, 2L, 2L, 2L, 3L, 3L, 3L, 3L, 
4L, 4L, 4L, 4L, 5L, 5L, 5L, 5L, 5L, 6L, 6L, 6L, 6L, 7L, 7L, 7L, 
7L, 8L, 8L, 8L, 8L, 9L, 9L, 9L, 9L, 9L, 10L, 10L, 10L, 10L, 11L, 
11L, 11L, 11L, 11L, 12L, 12L, 12L, 1L, 1L, 1L, 1L, 2L, 2L, 2L, 
2L, 3L, 3L, 3L, 3L, 3L, 4L, 4L, 4L, 4L, 5L, 5L, 5L, 6L, 6L, 6L, 
6L, 6L), survey_start = c("", "", "", "", "", "", "", "", "", 
"", "", "", "", "", "", "", "", "Y", "", "", "", "", "", "", 
"", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", 
"", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", 
"", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", 
"", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", 
"", "", "", "", "", "", "", "", "", "", "", "", "", "", "", "", 
"", "", "", "", "", "", "Y", "", "", "", "", "", "", "", "", 
"", "", "", "", "", "")), class = "data.frame", row.names = c(NA, 
-125L), .Names = c("study", "studyno", "date", "year", "month", 
"survey_start"))

代码1失败：

 trialData <- trialData %>% arrange(studyno, date) %>% group_by(studyno) %>%
mutate(survey_year = if_else(date >= date[survey_start == "Y" & year == 2007 & study == "site_1"][1] & date < date[month == 5 & year == 2008 & study == "site_1"][1], "survey_2007",
                     if_else(date >= date[survey_start == "Y" & year == 2008 & study == "site_1"][1] & date < date[month == 4 & year == 2009 & study == "site_1"][1], "survey_2008",
                     if_else(date >= date[survey_start == "Y" & year == 2009 & study == "site_1"][1] & date < date[month == 5 & year == 2010 & study == "site_1"][1], "survey_2009",
                     if_else(date >= date[survey_start == "Y" & year == 2010 & study == "site_1"][1] & date < date[month == 5 & year == 2011 & study == "site_1"][1], "survey_2010",
                     if_else(date >= date[survey_start == "Y" & year == 2011 & study == "site_1"][1] & date < date[month == 4 & year == 2012 & study == "site_1"][1], "survey_2011",
                     if_else(date >= date[survey_start == "Y" & year == 2012 & study == "site_1"][1] & date < date[month == 4 & year == 2013 & study == "site_1"][1], "survey_2012",
                     if_else(date >= date[survey_start == "Y" & year == 2013 & study == "site_1"][1] & date < date[month == 4 & year == 2014 & study == "site_1"][1], "survey_2013",
                     if_else(date >= date[survey_start == "Y" & year == 2014 & study == "site_1"][1] & date < date[month == 4 & year == 2015 & study == "site_1"][1], "survey_2014",
                     if_else(date >= date[survey_start == "Y" & year == 2015 & study == "site_1"][1] & date < date[month == 3 & year == 2016 & study == "site_1"][1], "survey_2015",        
                     if_else(date >= date[survey_start == "Y" & year == 2016 & study == "site_1"][1], "survey_2016","")))))))))))

代码2有效：

    trialData <- trialData %>% arrange(studyno, date) %>% group_by(studyno) %>%
  mutate(survey_year = if_else(date >= date[survey_start == "Y" & year == 2016 & study == "site_1"][1]                                                               , "survey_2016",
                           if_else(date >= date[survey_start == "Y" & year == 2014 & study == "site_1"][1] & date < date[month == 4 & year == 2015 & study == "site_1"][1], "survey_2014",
                           if_else(date >= date[survey_start == "Y" & year == 2013 & study == "site_1"][1] & date < date[month == 4 & year == 2014 & study == "site_1"][1], "survey_2013",
                           if_else(date >= date[survey_start == "Y" & year == 2012 & study == "site_1"][1] & date < date[month == 4 & year == 2013 & study == "site_1"][1], "survey_2012",
                           if_else(date >= date[survey_start == "Y" & year == 2011 & study == "site_1"][1] & date < date[month == 4 & year == 2012 & study == "site_1"][1], "survey_2011",
                           if_else(date >= date[survey_start == "Y" & year == 2010 & study == "site_1"][1] & date < date[month == 5 & year == 2011 & study == "site_1"][1], "survey_2010",
                           if_else(date >= date[survey_start == "Y" & year == 2009 & study == "site_1"][1] & date < date[month == 5 & year == 2010 & study == "site_1"][1], "survey_2009",
                           if_else(date >= date[survey_start == "Y" & year == 2008 & study == "site_1"][1] & date < date[month == 4 & year == 2009 & study == "site_1"][1], "survey_2008",
                           if_else(date >= date[survey_start == "Y" & year == 2007 & study == "site_1"][1] & date < date[month == 5 & year == 2008 & study == "site_1"][1], "survey_2007",""))))))))))

1 个答案:

答案 0 :(得分：2)

正如@akrun评论的那样，您可以通过合并数据而不是使用if_else来实现此目的。这个过程就是这样的：

  1. 创建仅包含启动调查年度的访问的数据集。
    • 定义开始和结束日期以及此处的调查年份标签
  2. 将起始访问数据加入原始数据
    • 保留属于调查年度的行
    • 仅选择识别访问所需的列和调查年份标签
  3. 将结果加回原始数据。

    4. 以下是使用dplyr：

    进行此操作的方法 
    library(tidyverse)
library(lubridate)

# Modify the data so that there's an overlap of survey years,
# in order to demonstrate how to deal with it
df <- as_tibble(trialData) %>% 
  mutate(survey_start = if_else(row_number() == 52, "Y", survey_start))

# Pick out rows that start a "survey year"
starts <- df %>% 
  filter(survey_start == "Y") %>% 
  group_by(study, studyno) %>% 
  transmute(
    survey_year = str_c("survey_", year),
    start_date = date,
    end_date   = pmin(
      start_date + years(1),  # make sure that the survey year
      lead(start_date),       # ends before next one starts
      na.rm = T
    )
  ) %>% ungroup()
#> Adding missing grouping variables: `study`, `studyno`

# Join all starts to the visit data
years <- df %>% 
  left_join(starts) %>% 
  # Keep rows which fall within one year of a start
  filter(date >= start_date, date < end_date) %>% 
  select(study, studyno, date, survey_year)
#> Joining, by = c("study", "studyno")

    现在years包含所有属于“调查年度”的访问

    # Join the year classifications to the original data
result <- df %>%
  left_join(years)
#> Joining, by = c("study", "studyno", "date")
stopifnot(nrow(result) == nrow(df))

    我们也可以查看结果：

    # Check the rows before and after each start
i <- which(result$survey_start == "Y")
result %>% slice(sort(c(i - 1, i, i + 1)))
#> # A tibble: 9 x 7
#>   study  studyno date        year month survey_start survey_year
#>   <chr>  <chr>   <date>     <int> <int> <chr>        <chr>      
#> 1 site_1 child_1 2014-05-01  2014     5 ""           <NA>       
#> 2 site_1 child_1 2014-05-05  2014     5 Y            survey_2014
#> 3 site_1 child_1 2014-05-13  2014     5 ""           survey_2014
#> 4 site_1 child_1 2015-01-09  2015     1 ""           survey_2014
#> 5 site_1 child_1 2015-01-17  2015     1 Y            survey_2015
#> 6 site_1 child_1 2015-01-19  2015     1 ""           survey_2015
#> 7 site_1 child_1 2016-03-07  2016     3 ""           <NA>       
#> 8 site_1 child_1 2016-03-17  2016     3 Y            survey_2016
#> 9 site_1 child_1 2016-03-24  2016     3 ""           survey_2016

    由reprex package（v0.2.0）创建于2018-02-22。

相关问题
最新问题