根据 R 中的概率表/数据框分配随机变量

Question

我有一个如下所示的概率数据框，名为 ptable：

 unique_id color share 
         1   red  0.3  
         1  blue  0.7  
         2   red  0.4  
         3  blue  0.5  

我想根据中的 share 变量随机分配一个 color 变量可能是另一个数据框 join_table 的表，如下所示。

unique_id count
         1    3  
         2    4 

我理解sample()，但我不知道如何通过共享unique_id分配概率。我最近的尝试是

join_table %>% 
group_by(unqiue_id) %>% 
mutate(color= sample(ptable$race[unique_id==ptable$unique_id], 
                     size=n(), 
                     prob=ptable$share[nique_id==ptable$unique_id], 
                     replace=TRUE))

任何帮助都会很棒。

Answer 1

代码中有两个拼写错误：

group_by(unqiue_id) 应该是 group_by(unique_id)
并且

prob=ptable$share[nique_id==ptable$unique_id] 应为 prob=ptable$share[unique_id==ptable$unique_id]。

这应该有效：

library(dplyr)
#> 
#> Attaching package: 'dplyr'
#> The following objects are masked from 'package:stats':
#> 
#>     filter, lag
#> The following objects are masked from 'package:base':
#> 
#>     intersect, setdiff, setequal, union
ptable <- tibble::tribble(
  ~unique_id, ~color, ~share,
1,   "red",  0.3,  
1,  "blue",  0.7, 
2,   "red",  0.4,
3,  "blue",  0.5)

join_table <- tibble::tribble(
  ~unique_id, ~count,
  1,    3,  
  2,    4)

join_table %>% 
  group_by(unique_id) %>% 
  mutate(color= sample(ptable$color[unique_id==ptable$unique_id], 
                       size=n(), 
                       prob=ptable$share[unique_id==ptable$unique_id], 
                       replace=TRUE))
#> # A tibble: 2 × 3
#> # Groups:   unique_id [2]
#>   unique_id count color
#>       <dbl> <dbl> <chr>
#> 1         1     3 blue 
#> 2         2     4 red

^{由 reprex 包 (v2.0.1)于 2022 年 3 月 1 日创建< /sup>}