Notebooks: Translate errorbars horizontally so they dont overlap

Author: Daniel Incicau

notebooks/layer5_CC_CS_connection.ipynb

@@ -9,6 +9,7 @@
    "source": [
     "from brian2 import *\n",
     "from matplotlib import pyplot as plt\n",
+    "from matplotlib.transforms import Affine2D\n",
     "import pandas as pd\n",
     "import numpy as np\n",
     "from src.nb_helpers import *\n",
@@ -803,6 +804,11 @@
     "fig, axs = plt.subplots(1, 3, figsize=(26, 6))\n",
     "xticks = np.linspace(0, len(cc_cs_weights)-1, len(cc_cs_weights))\n",
     "\n",
+    "trans1 = Affine2D().translate(-0.2, 0.0)\n",
+    "trans2 = Affine2D().translate(-0.1, 0.0)\n",
+    "trans3 = Affine2D().translate(+0.0, 0.0)\n",
+    "trans4 = Affine2D().translate(+0.1, 0.0)\n",
+    "\n",
     "os_cs_mean = [os_record[0] for os_record in os_mean_data]\n",
     "os_cs_std = [os_record[0] for os_record in os_std_data]\n",
     "os_cc_mean = [os_record[1] for os_record in os_mean_data]\n",
@@ -813,10 +819,10 @@
     "os_pv_std = [os_record[3] for os_record in os_std_data]\n",
     "\n",
     "# Plot orientation selectivity\n",
-    "axs[0].errorbar(xticks, os_cs_mean, fmt='b', yerr=os_cs_std, label='CS')\n",
-    "axs[0].errorbar(xticks, os_cc_mean, fmt='r', yerr=os_cc_std, label='CC')\n",
-    "axs[0].errorbar(xticks, os_sst_mean, fmt='g', yerr=os_sst_std, label='SST')\n",
-    "axs[0].errorbar(xticks, os_pv_mean, fmt='y', yerr=os_pv_std, label='PV')\n",
+    "axs[0].errorbar(xticks, os_cs_mean, fmt='b', yerr=os_cs_std, label='CS', transform=trans1+axs[0].transData, elinewidth=2)\n",
+    "axs[0].errorbar(xticks, os_cc_mean, fmt='r', yerr=os_cc_std, label='CC', transform=trans2+axs[0].transData, elinewidth=2)\n",
+    "axs[0].errorbar(xticks, os_sst_mean, fmt='g', yerr=os_sst_std, label='SST', transform=trans3+axs[0].transData, elinewidth=2)\n",
+    "axs[0].errorbar(xticks, os_pv_mean, fmt='y', yerr=os_pv_std, label='PV', transform=trans4+axs[0].transData, elinewidth=2)\n",
     "axs[0].set_xticks(xticks)\n",
     "axs[0].set_xticklabels(cc_cs_weights)\n",
     "axs[0].set_xlabel('Scalar of CC_CS connection')\n",
@@ -834,10 +840,10 @@
     "os_paper_pv_std = [os_record[3] for os_record in os_paper_std_data]\n",
     "\n",
     "# Plot orientation selectivity (paper)\n",
-    "axs[1].errorbar(xticks, os_paper_cs_mean, fmt='b', yerr=os_paper_cs_std, label='CS')\n",
-    "axs[1].errorbar(xticks, os_paper_cc_mean, fmt='r', yerr=os_paper_cc_std, label='CC')\n",
-    "axs[1].errorbar(xticks, os_paper_sst_mean, fmt='g', yerr=os_paper_sst_std, label='SST')\n",
-    "axs[1].errorbar(xticks, os_paper_pv_mean, fmt='y', yerr=os_paper_pv_std, label='PV')\n",
+    "axs[1].errorbar(xticks, os_paper_cs_mean, fmt='b', yerr=os_paper_cs_std, label='CS', transform=trans1+axs[1].transData, elinewidth=2)\n",
+    "axs[1].errorbar(xticks, os_paper_cc_mean, fmt='r', yerr=os_paper_cc_std, label='CC', transform=trans2+axs[1].transData, elinewidth=2)\n",
+    "axs[1].errorbar(xticks, os_paper_sst_mean, fmt='g', yerr=os_paper_sst_std, label='SST', transform=trans3+axs[1].transData, elinewidth=2)\n",
+    "axs[1].errorbar(xticks, os_paper_pv_mean, fmt='y', yerr=os_paper_pv_std, label='PV', transform=trans4+axs[1].transData, elinewidth=2)\n",
     "axs[1].set_xticks(xticks)\n",
     "axs[1].set_xticklabels(cc_cs_weights)\n",
     "axs[1].set_xlabel('Scalar of CC_CS connection')\n",
@@ -855,10 +861,10 @@
     "ds_pv_std = [ds_record[3] for ds_record in ds_std_data]\n",
     "\n",
     "# Plot direction selectivity\n",
-    "axs[2].errorbar(xticks, ds_cs_mean, fmt='b', yerr=ds_cs_std, label='CS')\n",
-    "axs[2].errorbar(xticks, ds_cc_mean, fmt='r', yerr=ds_cc_std, label='CC')\n",
-    "axs[2].errorbar(xticks, ds_sst_mean, fmt='g', yerr=ds_sst_std, label='SST')\n",
-    "axs[2].errorbar(xticks, ds_pv_mean, fmt='y', yerr=ds_pv_std, label='PV')\n",
+    "axs[2].errorbar(xticks, ds_cs_mean, fmt='b', yerr=ds_cs_std, label='CS', transform=trans1+axs[2].transData, elinewidth=2)\n",
+    "axs[2].errorbar(xticks, ds_cc_mean, fmt='r', yerr=ds_cc_std, label='CC', transform=trans2+axs[2].transData, elinewidth=2)\n",
+    "axs[2].errorbar(xticks, ds_sst_mean, fmt='g', yerr=ds_sst_std, label='SST', transform=trans3+axs[2].transData, elinewidth=2)\n",
+    "axs[2].errorbar(xticks, ds_pv_mean, fmt='y', yerr=ds_pv_std, label='PV', transform=trans4+axs[2].transData, elinewidth=2)\n",
     "axs[2].set_xticks(xticks)\n",
     "axs[2].set_xticklabels(cc_cs_weights)\n",
     "axs[2].set_xlabel('Scalar of CC_CS connection')\n",
@@ -877,6 +883,9 @@
     "fig, axs = plt.subplots(1, 2, figsize=(18, 6))\n",
     "xticks = np.linspace(0, len(cc_cs_weights)-1, len(cc_cs_weights))\n",
     "\n",
+    "trans1 = Affine2D().translate(-0.05, 0.0)\n",
+    "trans2 = Affine2D().translate(+0.0, 0.0)\n",
+    "\n",
     "firing_rates_mean_over_input_degrees = np.mean(firing_rates_mean_over_simulations, axis=0)\n",
     "firing_rates_std_over_input_degrees = np.mean(firing_rates_std_over_simulations, axis=0)\n",
     "\n",
@@ -890,10 +899,10 @@
     "fire_rate_pv_std = [fire_rate_std_by_weight[3] for fire_rate_std_by_weight in firing_rates_std_over_input_degrees]\n",
     "\n",
     "# Plot firing rate\n",
-    "axs[0].errorbar(xticks, fire_rate_cs_mean, fmt='b', yerr=fire_rate_cs_std, label='CS')\n",
-    "axs[0].errorbar(xticks, fire_rate_cc_mean, fmt='r', yerr=fire_rate_cc_std, label='CC')\n",
-    "axs[0].errorbar(xticks, fire_rate_sst_mean, fmt='g', yerr=fire_rate_sst_std, label='SST')\n",
-    "axs[0].errorbar(xticks, fire_rate_pv_mean, fmt='y', yerr=fire_rate_pv_std, label='PV')\n",
+    "axs[0].errorbar(xticks, fire_rate_cs_mean, fmt='b', yerr=fire_rate_cs_std, label='CS', elinewidth=2)\n",
+    "axs[0].errorbar(xticks, fire_rate_cc_mean, fmt='r', yerr=fire_rate_cc_std, label='CC', elinewidth=2)\n",
+    "axs[0].errorbar(xticks, fire_rate_sst_mean, fmt='g', yerr=fire_rate_sst_std, label='SST', elinewidth=2)\n",
+    "axs[0].errorbar(xticks, fire_rate_pv_mean, fmt='y', yerr=fire_rate_pv_std, label='PV', elinewidth=2)\n",
     "axs[0].set_xticks(xticks)\n",
     "axs[0].set_xticklabels(cc_cs_weights)\n",
     "axs[0].set_xlabel('Scalar of CC_CS connection')\n",
@@ -902,9 +911,9 @@
     "axs[0].legend(loc='best')\n",
     "\n",
     "# Plot os_rel, os_paper_rel and ds_rel\n",
-    "axs[1].errorbar(xticks, os_rel_mean_data, yerr=os_rel_std_data, label='OS rel')\n",
+    "axs[1].errorbar(xticks, os_rel_mean_data, yerr=os_rel_std_data, label='OS rel', transform=trans1+axs[1].transData, elinewidth=2)\n",
     "# axs[1].errorbar(xticks, os_paper_rel_mean_data, yerr=os_paper_rel_std_data, label='OS Paper rel')\n",
-    "axs[1].errorbar(xticks, ds_rel_mean_data, yerr=ds_rel_std_data, label='DS rel')\n",
+    "axs[1].errorbar(xticks, ds_rel_mean_data, yerr=ds_rel_std_data, label='DS rel', transform=trans2+axs[1].transData, elinewidth=2)\n",
     "axs[1].set_xticks(xticks)\n",
     "axs[1].set_xticklabels(cc_cs_weights)\n",
     "axs[1].set_xlabel('Scalar of CC_CS connection')\n",

notebooks/layer5_SST_Soma_selectivity.ipynb

@@ -9,6 +9,7 @@
    "source": [
     "from brian2 import *\n",
     "from matplotlib import pyplot as plt\n",
+    "from matplotlib.transforms import Affine2D\n",
     "import pandas as pd\n",
     "import numpy as np\n",
     "from src.nb_helpers import *\n",
@@ -772,6 +773,11 @@
     "fig, axs = plt.subplots(1, 3, figsize=(26, 6))\n",
     "xticks = np.linspace(0, len(sst_soma_conn_weights)-1, len(sst_soma_conn_weights))\n",
     "\n",
+    "trans1 = Affine2D().translate(-0.2, 0.0)\n",
+    "trans2 = Affine2D().translate(-0.1, 0.0)\n",
+    "trans3 = Affine2D().translate(+0.0, 0.0)\n",
+    "trans4 = Affine2D().translate(+0.1, 0.0)\n",
+    "\n",
     "os_cs_mean = [os_record[0] for os_record in os_mean_data]\n",
     "os_cs_std = [os_record[0] for os_record in os_std_data]\n",
     "os_cc_mean = [os_record[1] for os_record in os_mean_data]\n",
@@ -782,10 +788,10 @@
     "os_pv_std = [os_record[3] for os_record in os_std_data]\n",
     "\n",
     "# Plot orientation selectivity\n",
-    "axs[0].errorbar(xticks, os_cs_mean, fmt='b', yerr=os_cs_std, label='CS')\n",
-    "axs[0].errorbar(xticks, os_cc_mean, fmt='r', yerr=os_cc_std, label='CC')\n",
-    "axs[0].errorbar(xticks, os_sst_mean, fmt='g', yerr=os_sst_std, label='SST')\n",
-    "axs[0].errorbar(xticks, os_pv_mean, fmt='y', yerr=os_pv_std, label='PV')\n",
+    "axs[0].errorbar(xticks, os_cs_mean, fmt='b', yerr=os_cs_std, label='CS', transform=trans1+axs[0].transData, elinewidth=2)\n",
+    "axs[0].errorbar(xticks, os_cc_mean, fmt='r', yerr=os_cc_std, label='CC', transform=trans2+axs[0].transData, elinewidth=2)\n",
+    "axs[0].errorbar(xticks, os_sst_mean, fmt='g', yerr=os_sst_std, label='SST', transform=trans3+axs[0].transData, elinewidth=2)\n",
+    "axs[0].errorbar(xticks, os_pv_mean, fmt='y', yerr=os_pv_std, label='PV', transform=trans4+axs[0].transData, elinewidth=2)\n",
     "axs[0].set_xticks(xticks)\n",
     "axs[0].set_xticklabels(sst_soma_conn_weights)\n",
     "axs[0].set_xlabel('Scalar of SST->Soma connection')\n",
@@ -803,10 +809,10 @@
     "os_paper_pv_std = [os_record[3] for os_record in os_paper_std_data]\n",
     "\n",
     "# Plot orientation selectivity (paper)\n",
-    "axs[1].errorbar(xticks, os_paper_cs_mean, fmt='b', yerr=os_paper_cs_std, label='CS')\n",
-    "axs[1].errorbar(xticks, os_paper_cc_mean, fmt='r', yerr=os_paper_cc_std, label='CC')\n",
-    "axs[1].errorbar(xticks, os_paper_sst_mean, fmt='g', yerr=os_paper_sst_std, label='SST')\n",
-    "axs[1].errorbar(xticks, os_paper_pv_mean, fmt='y', yerr=os_paper_pv_std, label='PV')\n",
+    "axs[1].errorbar(xticks, os_paper_cs_mean, fmt='b', yerr=os_paper_cs_std, label='CS', transform=trans1+axs[1].transData, elinewidth=2)\n",
+    "axs[1].errorbar(xticks, os_paper_cc_mean, fmt='r', yerr=os_paper_cc_std, label='CC', transform=trans2+axs[1].transData, elinewidth=2)\n",
+    "axs[1].errorbar(xticks, os_paper_sst_mean, fmt='g', yerr=os_paper_sst_std, label='SST', transform=trans3+axs[1].transData, elinewidth=2)\n",
+    "axs[1].errorbar(xticks, os_paper_pv_mean, fmt='y', yerr=os_paper_pv_std, label='PV', transform=trans4+axs[1].transData, elinewidth=2)\n",
     "axs[1].set_xticks(xticks)\n",
     "axs[1].set_xticklabels(sst_soma_conn_weights)\n",
     "axs[1].set_xlabel('Scalar of SST->Soma connection')\n",
@@ -824,10 +830,10 @@
     "ds_pv_std = [ds_record[3] for ds_record in ds_std_data]\n",
     "\n",
     "# Plot direction selectivity\n",
-    "axs[2].errorbar(xticks, ds_cs_mean, fmt='b', yerr=ds_cs_std, label='CS')\n",
-    "axs[2].errorbar(xticks, ds_cc_mean, fmt='r', yerr=ds_cc_std, label='CC')\n",
-    "axs[2].errorbar(xticks, ds_sst_mean, fmt='g', yerr=ds_sst_std, label='SST')\n",
-    "axs[2].errorbar(xticks, ds_pv_mean, fmt='y', yerr=ds_pv_std, label='PV')\n",
+    "axs[2].errorbar(xticks, ds_cs_mean, fmt='b', yerr=ds_cs_std, label='CS', transform=trans1+axs[2].transData, elinewidth=2)\n",
+    "axs[2].errorbar(xticks, ds_cc_mean, fmt='r', yerr=ds_cc_std, label='CC', transform=trans2+axs[2].transData, elinewidth=2)\n",
+    "axs[2].errorbar(xticks, ds_sst_mean, fmt='g', yerr=ds_sst_std, label='SST', transform=trans3+axs[2].transData, elinewidth=2)\n",
+    "axs[2].errorbar(xticks, ds_pv_mean, fmt='y', yerr=ds_pv_std, label='PV', transform=trans4+axs[2].transData, elinewidth=2)\n",
     "axs[2].set_xticks(xticks)\n",
     "axs[2].set_xticklabels(sst_soma_conn_weights)\n",
     "axs[2].set_xlabel('Scalar of SST->Soma connection')\n",
@@ -846,6 +852,9 @@
     "fig, axs = plt.subplots(1, 2, figsize=(18, 6))\n",
     "xticks = np.linspace(0, len(sst_soma_conn_weights)-1, len(sst_soma_conn_weights))\n",
     "\n",
+    "trans1 = Affine2D().translate(-0.1, 0.0)\n",
+    "trans2 = Affine2D().translate(+0.0, 0.0)\n",
+    "\n",
     "firing_rates_mean_over_input_degrees = np.mean(firing_rates_mean_over_simulations, axis=0)\n",
     "firing_rates_std_over_input_degrees = np.mean(firing_rates_std_over_simulations, axis=0)\n",
     "\n",
@@ -871,9 +880,9 @@
     "axs[0].legend(loc='best')\n",
     "\n",
     "# Plot os_rel, os_paper_rel and ds_rel\n",
-    "axs[1].errorbar(xticks, os_rel_mean_data, yerr=os_rel_std_data, label='OS rel')\n",
+    "axs[1].errorbar(xticks, os_rel_mean_data, yerr=os_rel_std_data, label='OS rel', transform=trans1+axs[1].transData, elinewidth=2)\n",
     "# axs[1].errorbar(xticks, os_paper_rel_mean_data, yerr=os_paper_rel_std_data, label='OS Paper rel')\n",
-    "axs[1].errorbar(xticks, ds_rel_mean_data, yerr=ds_rel_std_data, label='DS rel')\n",
+    "axs[1].errorbar(xticks, ds_rel_mean_data, yerr=ds_rel_std_data, label='DS rel', transform=trans2+axs[1].transData, elinewidth=2)\n",
     "axs[1].set_xticks(xticks)\n",
     "axs[1].set_xticklabels(sst_soma_conn_weights)\n",
     "axs[1].set_xlabel('Scalar of SST->Soma connection')\n",