The ggswim package provides a convenient set of commands to easily create swimmer plots. As an extension of ggplot2, it streamlines the process of generating legends that effectively communicate events of interest along subject response paths.
ggswim solves some of the headaches associated with layer management in ggplot2 by leveraging the ggnewscale package and presenting an optimized workflow to get a swimmer plot.
You can install the development version of ggswim like so:
devtools::install_github("CHOP-CGTInformatics/ggswim")
To help you get started, ggswim includes three sample datasets:
patient_data
, infusion_events
, and end_study_events
. These
de-identified datasets simulate real world data related to infusions,
disease assessments, and study statuses for a clinical trial.
ggswim offers several geom-functions, and by using geom_swim_lane()
we
can set up the horizontal response paths of our swimmer plot, i.e. the
“lanes”. We’ll also set up corresponding arrows to indicate subjects
that are still on the trial:
library(ggswim)
library(ggplot2)
# Construct arrow_data for arrow display later
arrow_data <- patient_data |>
dplyr::left_join(
end_study_events |>
dplyr::select(pt_id, end_study_name),
by = "pt_id"
) |>
dplyr::select(pt_id, end_time, end_study_name) |>
dplyr::filter(.by = pt_id, end_time == max(end_time)) |>
unique()
p <- patient_data |>
ggplot() +
geom_swim_lane(
mapping = aes(
x = start_time, y = pt_id, xend = end_time,
color = disease_assessment
)
) +
geom_swim_arrow(
data = arrow_data,
mapping = aes(xend = end_time, y = pt_id)
) +
scale_color_brewer(
name = "Overall Disease Assessment",
palette = "Set1"
)
p
Next we’ll add on events of interest: end of study updates and
infusions. We’ll refer to these as “markers” and call them with two more
“geom” functions: geom_swim_point()
and geom_swim_label()
.
p <- p +
new_scale_color() +
geom_swim_point(
data = infusion_events,
aes(x = time_from_initial_infusion, y = pt_id, color = infusion_type),
size = 5
) +
geom_swim_label(
data = end_study_events,
aes(x = time_from_initial_infusion, y = pt_id, label_vals = end_study_label, label_names = end_study_name),
label.size = NA, fill = NA, size = 5
)
p
Finally, we’ll beautify the plot with familiar ggplot2 techniques and a
last finishing touch with theme_ggswim()
:
p +
scale_color_brewer(name = "Markers", palette = "Set2") +
labs(title = "My Swimmer Plot") +
xlab("Time Since Infusion (Months)") + ylab("Patient ID") +
theme_ggswim()
We invite you to give feedback and collaborate with us! If you are familiar with GitHub and R packages, please feel free to submit a pull request. Please do let us know if ggswim fails for whatever reason with your use case and submit a bug report by creating a GitHub issue.
Please note that this project is released with a Contributor Code of Conduct. By participating you agree to abide by its terms.