R/two_p_summarize.R

In TeachBayes: Teaching Bayesian Inference

two_p_summarize <- function(prob_matrix){
  p1 <- as.numeric(dimnames(prob_matrix)[[1]])
  p2 <- as.numeric(dimnames(prob_matrix)[[2]])
  P1 <- rep(p1, length(p2))
  P2 <- rep(p2, each=length(p1))
  PROB <- as.vector(prob_matrix)
  diff21 <- factor(P2 - P1)
  df <- data.frame(P1, P2, PROB, diff21)
  df <- summarize(group_by(df, diff21), Prob=sum(PROB))
  mutate(df, diff21 = as.numeric(as.character(diff21)))
}