fix: notebook 4, receptor

FZJ-INM1-BDA · Oct 10, 2024 · 07cb0e9 · 07cb0e9
1 parent 1ddb25c
commit 07cb0e9
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/paper_notebooks/4-probabilistic-assignment.ipynb b/paper_notebooks/4-probabilistic-assignment.ipynb
@@ -202,9 +202,9 @@
    "source": [
     "region = pmaps.get_region(regionname) \n",
     "profiling = [\n",
-    "    [siibra.modality_vocab.modality.NEUROTRANSMITTER_RECEPTOR_DENSITY, {}],\n",
-    "    [siibra.modality_vocab.modality.CELL_BODY_DENSITY, {}],\n",
-    "    # [siibra.modality_vocab.modality.STREAMLINECOUNTS, {}],\n",
+    "    [siibra.modality_vocab.modality.NEUROTRANSMITTER_RECEPTOR_DENSITY, \"fingerprint\" {}],\n",
+    "    [siibra.modality_vocab.modality.CELL_BODY_DENSITY, None, {}],\n",
+    "    # [siibra.modality_vocab.modality.STREAMLINECOUNTS, None, {}],\n",
     "]"
    ]
   },
@@ -217,9 +217,16 @@
    "source": [
     "fig, axs = plt.subplots(len(profiling), 1, figsize=(6.0, 2.5 * len(profiling)))\n",
     "\n",
-    "for i, (modality, kwargs) in enumerate(profiling):\n",
+    "for i, (modality, categorization, kwargs) in enumerate(profiling):\n",
     "    features = siibra.find_features(region, modality, **kwargs)\n",
-    "    features[0].plot(ax=axs[i])\n",
+    "    if categorization is None:\n",
+    "        feature = features[0]\n",
+    "    else:\n",
+    "        for f in features:\n",
+    "            if categorization in f.categorizations:\n",
+    "                feature = f\n",
+    "                break\n",
+    "    feature.plot(ax=axs[i])\n",
     "    \n",
     "plt.tight_layout()"
    ]