6.5 Coefficient plots: Coloring

• Figure 6.3 below provides a coefficient plot and colors estimations for subsets.

Figure 6.3: Coefficient plot: Coloring

6.5.1 Lab: Data & code

The logic of estimating the models is the same as in the section [Coefficient plots: Facetting]. The only difference is that we use another variable namely Infant.Mortality_cat.

swiss <- swiss %>% 
         mutate(Infant.Mortality_cat = cut(Infant.Mortality, 
                                  breaks = quantile(Infant.Mortality, probs = seq(0, 1, 0.5)),
                                  labels = c("low", "high")))

#table(swiss$Infant.Mortality, swiss$Infant.Mortality_cat)
results <- swiss %>% 
  filter(!is.na(Infant.Mortality_cat)) %>%
  select(Infant.Mortality_cat, Fertility, Agriculture, Education, Catholic) %>%
  nest(data = c(Fertility, Agriculture, Education, Catholic)) %>% 
  mutate(fit = map(data, ~ lm(Fertility ~ Catholic + Agriculture + Education, data = .)),
         results = map(fit, tidy),
         results_90 = map(fit, conf.level = 0.90, confint_tidy),
         results_95 = map(fit, conf.level = 0.95, confint_tidy)) %>%
     mutate(results_90 = map(results_90, ~ rename_all(., function(x){paste(x, "_90", sep="")})),
            results_95 = map(results_95, ~ rename_all(., function(x){paste(x, "_95", sep="")}))) %>%
  unnest(c(results, results_90, results_95)) %>%
  rename(Variable = term,
                  Coefficient = estimate,
                  SE = std.error) %>%
  filter(Variable != "(Intercept)")

Plotting the data is straightforward again. We use the same code as above but now instead of facetting we simply specify colour = Infant.Mortality_cat within aes().

  # GGPLOT
  ggplot(results, aes(x = Variable, y = Coefficient, colour = Infant.Mortality_cat)) +
        geom_hline(yintercept = 0, colour = gray(1/2), lty = 2) +
        geom_point(aes(x = Variable, 
                    y = Coefficient), position=position_dodge(width=0.3)) + 
        geom_linerange(aes(x = Variable, 
                     ymin = conf.low_90,
                     ymax = conf.high_90),
                   lwd = 1, position=position_dodge(width=0.3)) +
        geom_linerange(aes(x = Variable, 
                     ymin = conf.low_95,
                     ymax = conf.high_95),
                   lwd = 1/2, position=position_dodge(width=0.3)) + 
        ggtitle("Outcome: Fertility (Subsets: Infant.Mortality)") +
        coord_flip()