From 0be00652e19dbd07c8bbcb64789aa91dc573f67e Mon Sep 17 00:00:00 2001
From: Jens Sauer <jsauer65@gmail.com>
Date: Thu, 19 Nov 2020 19:38:11 +0100
Subject: [PATCH] Visualize asct per diagnosis per age

This adds ggplots to visualize the number of ascts per diagnosis per
age in various methods:

* Histogram
* Density plot
* Frequency plot
* Scatter plot
---
 visualize.R | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/visualize.R b/visualize.R
index 06c3328..2b78a73 100644
--- a/visualize.R
+++ b/visualize.R
@@ -11,6 +11,33 @@ sma_hist_asct_age <- function() {
 	ggplot(data, aes(x = asct_age)) + geom_histogram(binwidth = 5)
 }
 
+i_asct_age_data <- function() {
+	return(data.frame(asct_age = secmal$asct_age, dx = secmal$diagnosis))
+}
+
+sma_hist_dx_age <- function() {
+	data <- i_asct_age_data()
+	ggplot(data, aes(x = asct_age, colour = dx, fill = dx)) +
+		geom_histogram(binwidth = 1)
+}
+
+sma_dens_dx_age <- function() {
+	data <- i_asct_age_data()
+	ggplot(data, aes(x = asct_age, colour = dx, fill = dx)) +
+		geom_density(alpha = 0.3)
+}
+sma_freq_dx_age <- function() {
+	data <- i_asct_age_data()
+	ggplot(data, aes(x = asct_age, colour = dx)) +
+		geom_freqpoly(binwidth = 1)
+}
+
+sma_jitt_dx_age <- function() {
+	data <- i_asct_age_data()
+	ggplot(data, aes(x = asct_age, y = dx, colour = dx, fill = dx)) +
+		geom_jitter(width = 0.4, height = 0.2)
+}
+
 sma_hist_asct_year <- function() {
 	data <- data.frame(asct_year = secmal$asct_year)
 	ggplot(data, aes(x = asct_year)) + geom_histogram(binwidth = 1)
@@ -51,4 +78,8 @@ sma_plot_file_hist <- function() {
 	sma_plot_file("dx_year_dens.png", png, sma_dens_dx_year)
 	sma_plot_file("dx_year_freq.png", png, sma_freq_dx_year)
 	sma_plot_file("dx_year_jitt.png", png, sma_jitt_dx_year)
+	sma_plot_file("dx_age_hist.png", png, sma_hist_dx_age)
+	sma_plot_file("dx_age_dens.png", png, sma_dens_dx_age)
+	sma_plot_file("dx_age_freq.png", png, sma_freq_dx_age)
+	sma_plot_file("dx_age_jitt.png", png, sma_jitt_dx_age)
 }