r语言 - 使用 dplyr、自定义函数或咕噜声的多个条件 if-else

Question

我有一个结构类似于以下内容的数据框：

set.seed(123)  
df<-data_frame(SectionName = rep(letters[1:2], 50),
TimeSpentSeconds = sample(0:360, 100, replace = TRUE),
Correct = sample(0:1, 100, replace = TRUE))

我想通过获取属于特定范围(小于 30、介于 30-60、介于 60-90 之间、...、大于 180(的所有 TimeSpentSeconds 值来总结此数据框，将时间标记为这些范围，按 SectionName 对它们进行分组，并找到 Correct 列的总和，以便生成的数据框看起来像这样(某些内容(：

TimeGroup             SectionName Correct
<fct>                 <chr>         <int>
1 LessThan30Secs        a                 2
2 LessThan30Secs        b                 3
3 30-60 Seconds         a                 4
4 30-60 Seconds         b                 3
5 60-90 Seconds         a                 2
6 60-90 Seconds         b                 3
7 90-120 Seconds        a                 4
8 90-120 Seconds        b                 0
9 120-150 Seconds       a                 4
10 120-150 Seconds       b                 0
11 150-180 Seconds       a                 1
12 150-180 Seconds       b                 2
13 GreaterThan180Seconds a                11
14 GreaterThan180Seconds b                11

我能够使用以下 if-else 代码成功做到这一点，其中我将所有时间都突变为具有适当标签、分组和汇总的新列：

x <- c("LessThan30Secs", "30-60 Seconds", "60-90 Seconds","90-120 Seconds", 
"120-150 Seconds", "150-180 Seconds", "GreaterThan180Seconds") 
df %>% 
mutate(TimeGroup = if_else(TimeSpentSeconds >= 0 & TimeSpentSeconds <= 30, "LessThan30Secs",
if_else(TimeSpentSeconds > 30 & TimeSpentSeconds <= 60, "30-60 Seconds",
if_else(TimeSpentSeconds > 60 & TimeSpentSeconds <= 90, "60-90 Seconds",
if_else(TimeSpentSeconds > 90 & TimeSpentSeconds <= 120, "90-120 Seconds",
if_else(TimeSpentSeconds > 120 & TimeSpentSeconds <= 150, "120-150 Seconds", 
if_else(TimeSpentSeconds > 150 & TimeSpentSeconds <= 180, "150-180 Seconds",
if_else(TimeSpentSeconds > 180, "GreaterThan180Seconds", "")))))))) %>%
mutate(TimeGroup = factor(TimeGroup, levels = x)) %>%
arrange(TimeGroup) %>%
group_by(TimeGroup, SectionName) %>%
summarise(Correct = sum(Correct))

但是，必须有更好的方法来做到这一点。我考虑过写一个函数，但没有走得太远，因为我不擅长写函数。

有没有人对通过我没有想到的 dplyr 方法完成相同输出的更优雅的方式有任何想法，编写一个自定义函数，可能在某个时候使用 purrr 包，或者其他一些 r 函数？

Answer 1

>case_when()会做你想做的事。它是嵌套ifelse()语句的整洁替代方案。

library(dplyr)
mutate(df,
TimeGroup = case_when(
TimeSpentSeconds <= 30 ~ "30 Seconds or less",
TimeSpentSeconds <= 60 ~ "31-60 Seconds",
TimeSpentSeconds <= 90 ~ "61-90 Seconds",
TimeSpentSeconds <= 120 ~ "91-120 Seconds",
TimeSpentSeconds <= 150 ~ "121-150 Seconds", 
TimeSpentSeconds <= 180 ~ "151-180 Seconds",
TimeSpentSeconds > 180 ~ "Greater Than 180 Seconds",
TRUE ~ NA_character_)
)

最后一个参数是不符合任何条件的记录的全部捕获，例如时间是否小于 0 秒。

Answer 2

我们可以使用cut(或findInterval(而不是多个嵌套的ifelse语句轻松做到这一点

lbls <- c('LessThan30secs', '30-60 Seconds', '60-90 Seconds', 
'90-120 Seconds', '120-150 Seconds', '150-180 Seconds', 'GreaterThan180Seconds')
df %>% 
group_by(TimeGroup = cut(TimeSpentSeconds, 
breaks = c(seq(0, 180, by = 30), Inf), labels = lbls), 
SectionName) %>%
summarise(Correct = sum(Correct)) %>%
na.omit

Answer 3

``` r
library(tidyverse)
set.seed(123)
df<-data_frame(SectionName = rep(letters[1:2], 50),
TimeSpentSeconds = sample(0:360, 100, replace = TRUE),
Correct = sample(0:1, 100, replace = TRUE))
time_spent_range <- function(value, start, end, interval) {
end <- end + (end%%interval) # make sure the end value is divisible by the interval
bins_start <- seq(start, end - interval, by = interval)
bins_end <- seq(start + interval, end, by = interval)
bins_tibble <- tibble(bin_start = bins_start, 
bin_end = bins_end) %>% 
mutate(in_bin = if_else((value > bin_start|(value == 0 & bin_start == 0)) 
& value <= bin_end,
1,
0)) %>% 
filter(in_bin == 1)
bin <- paste0(as.character(bins_tibble$bin_start[1]), 
'-', 
as.character(bins_tibble$bin_end[1]),
' Seconds')
return(bin)
}
df %>% 
mutate(TimeGroup = map_chr(TimeSpentSeconds, time_spent_range, start = 0, end = max(df$TimeSpentSeconds) , interval = 30))
#> # A tibble: 100 x 4
#>    SectionName TimeSpentSeconds Correct TimeGroup      
#>    <chr>                  <int>   <int> <chr>          
#>  1 a                        103       1 90-120 Seconds 
#>  2 b                        284       0 270-300 Seconds
#>  3 a                        147       0 120-150 Seconds
#>  4 b                        318       1 300-330 Seconds
#>  5 a                        339       0 330-360 Seconds
#>  6 b                         16       1 0-30 Seconds   
#>  7 a                        190       1 180-210 Seconds
#>  8 b                        322       1 300-330 Seconds
#>  9 a                        199       0 180-210 Seconds
#> 10 b                        164       0 150-180 Seconds
#> # ... with 90 more rows
```

创建于 2018-08-26 由 reprex 软件包 (v0.2.0(。