Displays scatter plot in notebook.

modules/analysis.py

@@ -166,4 +166,6 @@ def sc_compartment2embedding(embeds_path,data_dir,output_file="tutorial_scatterp
     ax.legend(handles=handles, labels=labels, bbox_to_anchor=(1.05, 1), loc=2, borderaxespad=0.)
     plt.tight_layout()
     plt.savefig(output_file, dpi=300)
-    plt.close('all')
+    plt.close('all')
+
+    return (vec,label,pal)

tutorial.ipynb

@@ -285,14 +285,24 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
+      "(4238, 5432)\n",
       "(4238, 5432)\n"
      ]
     }
    ],
    "source": [
     "from modules.analysis import sc_compartment2embedding\n",
+    "import seaborn as sns\n",
+    "import matplotlib.pyplot as plt\n",
     "\n",
-    "sc_compartment2embedding('./tutorial_embeds.hdf5','/mnt/e/data/pfc/','tutorial_scatter.pdf')"
+    "(vec,label,pal) = sc_compartment2embedding('./tutorial_embeds.hdf5','/mnt/e/data/pfc/','tutorial_scatter.pdf')\n",
+    "\n",
+    "fig, ax = plt.subplots(figsize=(7, 5))\n",
+    "sns.scatterplot(x=vec[:, 0], y=vec[:, 1], hue=label, linewidth=0, s=2, alpha=1.0, palette=pal)\n",
+    "\n",
+    "handles, labels = ax.get_legend_handles_labels()\n",
+    "labels, handles = zip(*sorted(zip(labels, handles), key=lambda t: t[0]))\n",
+    "ax.legend(handles=handles, labels=labels, bbox_to_anchor=(1.05, 1), loc=2, borderaxespad=0.)"
    ]
   }
  ],