dplyr case_when 在复杂场景中

Question

假设我在每轮研究中按国家、轮次和类型产生概率 table。而且，我需要根据一个人到那时为止参加的回合来计算权重。权重计算为所有概率总和 (p) 减去一个人参与的那轮所有概率的乘积的倒数。

我考虑过使用 case_when() 并且如果我找不到在未来几轮中自动执行它的方法，至少将其写出 10 轮，但不确定我的方法是否正确。感谢真正的 R 用户的任何指导！

For id=1 in the example below,
p is 0.78584735 for round=1 and type=2 and country="DE"
p is 0.07271288 for round=2 and type=2 and country="DE"
Then, p_tot should be (0.78584735+0.07271288)- (0.78584735*0.07271288)


# Table with probabilities

  set.seed(1245)
  prob_table <- data.frame(country=c(rep("DE",6), rep("UK",6)), 
                           round=c(rep(1,3),rep(2,3),rep(1,3),rep(2,3)),
                           type=c(rep(1:3,2)), p=c(runif(12)))

# Data frame with participants

df <- data.frame(id=c(1:15), country=c(rep("DE",8), rep("UK",7)), 
                 round=c(2,3,1,1,1,2,1,1,2,3,1,3,2,2,2),
                 type=c(2,3,1,1,1,2,3,1,2,1,1,3,1,1,2))

# Calculate total probability

df %<>% mutate(
  p_tot = case_when(
    country=="DE" & round==1 & type==1 
    ~ prob_table%>% filter(country=="DE" & round<=1 & type==1) %>% 
      sum(all elements of p column)-multiply(all elements of p column),

    country=="DE" & round==1 & type==1 
    ~ prob_table%>% filter(country=="DE" & round<=1 & type==1) %>% 
      sum(all elements of p column)-multiply(all elements of p column),

    ...
    ...

    TRUE ~ NA
  )
)


# calculate weight

df$weight <- 1/df$p_tot

Answer 1

您可以使用每一行的值来创建过滤器，而不是对其进行硬编码。

通常像这样的问题可以通过连接两个表来解决，但是 less than equal (round<=1) 条件使事情变得棘手，所以我使用了与您类似的方法。

希望这对您有所帮助：

library(dplyr)


# We change name to avoid collision during the filter
names(prob_table) <- paste('p', names(prob_table), sep = '_')

# Calculate total probability
df %>% 
  rowwise() %>% 
  mutate(
    p_tot = prob_table %>% 
      filter(p_country == country, p_round <= round, p_type == type) %>% 
      summarise(s = sum(p_p), 
                m = prod(p_p),
                f = s - m) %>% 
      pull(f),
    weight = 1 / p_tot
  )
#> Source: local data frame [15 x 6]
#> Groups: <by row>
#> 
#> # A tibble: 15 x 6
#>       id country round  type p_tot weight
#>    <int> <fct>   <dbl> <dbl> <dbl>  <dbl>
#>  1     1 DE          2     2 0.801   1.25
#>  2     2 DE          3     3 0.447   2.24
#>  3     3 DE          1     1 0     Inf   
#>  4     4 DE          1     1 0     Inf   
#>  5     5 DE          1     1 0     Inf   
#>  6     6 DE          2     2 0.801   1.25
#>  7     7 DE          1     3 0     Inf   
#>  8     8 DE          1     1 0     Inf   
#>  9     9 UK          2     2 0.532   1.88
#> 10    10 UK          3     1 0.475   2.10
#> 11    11 UK          1     1 0     Inf   
#> 12    12 UK          3     3 0.762   1.31
#> 13    13 UK          2     1 0.475   2.10
#> 14    14 UK          2     1 0.475   2.10
#> 15    15 UK          2     2 0.532   1.88

^{由 reprex package (v0.3.0)}

于 2020-06-17 创建

dplyr case_when 在复杂场景中

dplyr case_when in a complex scenario

r

case-when

dplyr