add legend fig 2

Author: MelissaGraham

DP1/300_Science_Demos/306_Extragalactic_transients/306_3_Cross_match_queries.ipynb

@@ -605,12 +605,19 @@
     "            dx = np.where(fsodo_table_2['diaObjectId'] == unique_dia_obj_id[x])\n",
     "            lc = fsodo_table_2[dx]\n",
     "            fx = np.where(lc['band'] == filt)[0]\n",
-    "            ax[i, j].plot(lc['expMidptMJD'][fx], lc['psfDiffFlux'][fx],\n",
-    "                          filter_symbols[filt], ms=5, mew=0, alpha=0.5,\n",
-    "                          color=filter_colors[filt])\n",
+    "            if (i == 0) & (j == 0):\n",
+    "                ax[i, j].plot(lc['expMidptMJD'][fx], lc['psfDiffFlux'][fx],\n",
+    "                              filter_symbols[filt], ms=5, mew=0, alpha=0.5,\n",
+    "                              color=filter_colors[filt], label=filt)\n",
+    "            else:\n",
+    "                ax[i, j].plot(lc['expMidptMJD'][fx], lc['psfDiffFlux'][fx],\n",
+    "                              filter_symbols[filt], ms=5, mew=0, alpha=0.5,\n",
+    "                              color=filter_colors[filt])\n",
     "            del dx, fx\n",
     "        ax[i, j].set_title(unique_dia_obj_id[x])\n",
     "        del lc\n",
+    "        if (i == 0) & (j == 0):\n",
+    "            ax[i, j].legend(loc='best', ncol=3, handletextpad=0)\n",
     "        if i == 2:\n",
     "            ax[i, j].xaxis.set_label_text('MJD (days)')\n",
     "        if j == 0:\n",