INFRA: qqplot

Add density plots when plotting qqplots to compare two data.
tskit-dev · Feb 12, 2024 · 7004b8a · 7004b8a
1 parent 44169f6
commit 7004b8a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/verification.py b/verification.py
@@ -71,6 +71,13 @@ def _plot_qq_compare(self, data1, data1_name, data2, data2_name):
         pyplot.savefig(f, dpi=72)
         pyplot.close("all")
 
+        sns.kdeplot(data1, color="b", fill=True, legend=False, label=data1_name)
+        sns.kdeplot(data2, color="r", fill=True, legend=False, label=data2_name)
+        pyplot.legend()
+        f = self._build_filename(data1_name, data2_name, "density_histogram")
+        pyplot.savefig(f, dpi=72)
+        pyplot.close("all")
+
 
 def model_list(loc, scale):
     df = 10