add heatmap example to chapter 10

2022-06-25 00:33:18 +02:00 · 2022-06-25 00:33:18 +02:00 · 85aa87d83b
commit 85aa87d83b
parent a43a20a507
1 changed files with 12 additions and 0 deletions
--- a/ch10.jl
+++ b/ch10.jl
@ -101,6 +101,18 @@ data

 aq2 = DataFrame(data)

+# Codes for section 10.1.4
+
+aq1
+
+using Statistics
+using StatsBase
+cor_mat = pairwise(cor, eachcol(aq1))
+
+using Plots
+heatmap(names(aq1), names(aq1), cor_mat;
+        aspect_ratio=:equal, xlim=(0,8))
+
 # Codes for listing 10.3

 data_dfs = map(DataFrame, data)