swo

import numpy as np


class MyRNG:
    def __init__(self, seed):
        # store the global state, so we can reset it
        global_state = np.random.get_state()
        # set the seed and get the state
        np.random.seed(seed)
        self.state = np.random.get_state()

swo

import polars as pl
import numpy as np

baseline_pars = {
    "N": int(1e3),
    "p_hospitalized": 0.10,
    "hosp_time_fun": lambda rng, size: rng.normal(loc=0.0, scale=1.0, size=size),
    "p_vax": 0.25,
    "vax_time_fun": lambda rng, size: rng.normal(loc=0.0, scale=1.0, size=size),
    "ve": 0.50,

swo

#' @param list_of_plots plots to save, one per page
#' @param path path to output pdf
#' @param ... additional arguments to `ggplot2::ggsave()`
save_multipage_pdf <- function(list_of_plots, path, ...) {
  # path must end with .pdf
  stopifnot(grepl("\\.pdf$", path))
  ggplot2::ggsave(
    path,
    gridExtra::marrangeGrob(list_of_plots, ncol = 1, nrow = 1)
    ...

swo

import polars as pl

# Making data frames ------------------------------------------
#  with dictionary
df = pl.DataFrame({
    'name': ['foo', 'bar', 'baz'],
    'bar': [0, 1, 2],
    'qux': [0.0, 1.0, 2.0]
})

swo

library(tidyverse)

# "faithful" is a dataset about a geyser, showing size of eruption vs. waiting
# time before the eruption
data <- as_tibble(datasets::faithful) %>%
  mutate(index = 1:n()) %>%
  head(10)

data

swo

sigfig <- function(x, n = 2) {
  formatC(signif(x, digits = n), digits = n, format = "fg", flag = "#")
}

swo

# Say I have some short-hand labels like "Ec/q",
# which actually stand for "E. coli & quinolones",
# but I want "E. coli" to be in italics

library(tidyverse)

n <- 100

tibble(
  x = rnorm(n),