在两个数据帧之间进行T检验,并按R

时间:2019-06-26 13:21:15

标签: r dataframe dplyr broom hypothesis-test

标题可能不是很清楚,但是希望我可以在这里更好地描述它。我有两个数据框,每个数据框描述了不同类型客户的每月支出。例如,对于A客户,我有一个数据框,例如

year_month    customer_id     monthly_spending
201301        123             5.50
201301        124             2.30
201301        125             6.80
201302        123             8.30
201302        124             5.60

然后我也为B客户提供了类似的数据框。理想情况下,我需要一个数据框,其中每个月都有T测试结果,用于比较A客户和B客户之间的支出。如果所有数据都在一个数据帧中,则可以使用dplyr()和Broom()进行此操作。如果我有两个数据框,有没有办法做到这一点?或者最好将两者合并在一起,然后再进行T检验和group_by year_month?

1 个答案:

答案 0 :(得分:0)

我在概念上的评论中同意@Gainz。合并数据框可能是最简单的方法。然而,真正的合并类型操作例如可能实际上不需要left_join()等;只需使用bind_rows()就足够了。

# read data from question
df_A <-
'year_month    customer_id     monthly_spending
201301        123             5.50
201301        124             2.30
201301        125             6.80
201302        123             8.30
201302        124             5.60' %>% 
    str_replace_all('[ ]++', '\t') %>% 
    read_tsv

# simulate more data for customer group "B"
df_B <- 
    df_A %>%    
    mutate(monthly_spending = monthly_spending + rnorm(5))

# combine the dataframes
df_all <-
    bind_rows(list('A' = df_A, 'B' = df_B), .id = 'customer_type')

# use broom to do a tidy t.test()
df_all %>% 
    group_by(year_month) %>%
    do(tidy(t.test(formula = monthly_spending ~ customer_type, 
                   data=.)))

这里的组合数据帧df_all

# A tibble: 10 x 4
   customer_type year_month customer_id monthly_spending
   <chr>              <dbl>       <dbl>            <dbl>
 1 A                 201301         123             5.5 
 2 A                 201301         124             2.3 
 3 A                 201301         125             6.8 
 4 A                 201302         123             8.3 
 5 A                 201302         124             5.6 
 6 B                 201301         123             6.25
 7 B                 201301         124             2.63
 8 B                 201301         125             7.04
 9 B                 201302         123             9.11
10 B                 201302         124             5.11

进行t.tests的结果是

# A tibble: 2 x 11
# Groups:   year_month [2]
  year_month estimate estimate1 estimate2 statistic p.value parameter conf.low
       <dbl>    <dbl>     <dbl>     <dbl>     <dbl>   <dbl>     <dbl>    <dbl>
1     201301  -0.692       4.87      5.56   -0.341    0.750      3.93    -6.35
2     201302  -0.0453      6.95      7.00   -0.0334   0.979      1.02   -16.4 
# … with 3 more variables: conf.high <dbl>, method <chr>, alternative <chr>