final improvements

Author: galaxyumi

DP1/300_Science_Demos/305_Galactic_variables_and_transients/305_2_Stellar_Variability_Characterization.ipynb

@@ -480,7 +480,7 @@
     "                             \"FROM dp1.ForcedSourceOnDiaObject as fsodo \"\n",
     "                             \"JOIN dp1.CcdVisit as cv \"\n",
     "                             \"ON (cv.visitId = fsodo.visit AND cv.detector = fsodo.detector)\"\n",
-    "                             \"WHERE fsodo.psfFlux > 0.0 AND diaObjectId = \"+str(id))\n",
+    "                             \"WHERE pixelFlags_bad = 0 AND diaObjectId = \"+str(id))\n",
     "\n",
     "    results = results.to_table()\n",
     "    results_dict[id] = results\n",
@@ -544,11 +544,17 @@
     "    obj_mjd_days = np.array(results['expMidptMJD']) * u.day\n",
     "    obj_fluxes = np.array(results['psfFlux'])\n",
     "    obj_flux_errs = np.array(results['psfFluxErr'])\n",
-    "    obj_frequency, obj_power =\\\n",
-    "        LombScargleMultiband(obj_mjd_days, obj_fluxes,\n",
-    "                             results['band'], obj_flux_errs).autopower(minimum_frequency=minfreq,\n",
-    "                                                                       maximum_frequency=maxfreq,\n",
-    "                                                                       samples_per_peak=20)\n",
+    "    ls_model = LombScargleMultiband(\n",
+    "        obj_mjd_days,\n",
+    "        obj_fluxes,\n",
+    "        results['band'],\n",
+    "        obj_flux_errs\n",
+    "    )\n",
+    "    obj_frequency, obj_power = ls_model.autopower(\n",
+    "        minimum_frequency=minfreq,\n",
+    "        maximum_frequency=maxfreq,\n",
+    "        samples_per_peak=20\n",
+    "    )\n",
     "    max_power = np.argmax(obj_power)\n",
     "    obj_freq = obj_frequency[max_power]\n",
     "    obj_period = 1.0 / obj_freq\n",
@@ -682,8 +688,8 @@
     "                 ms=8, mew=0, alpha=0.5, label=band)\n",
     "\n",
     "    plt.gca().invert_yaxis()\n",
-    "    plt.legend()\n",
-    "    plt.title(f\"{my_id}; period: {obj_period: .8F}\")\n",
+    "    plt.legend(ncol=4, loc=\"lower right\")\n",
+    "    plt.title(f\"Period: {obj_period: .8F}\")\n",
     "    plt.xlabel('phase')\n",
     "    plt.ylabel('magnitude')\n",
     "    plt.ylim(np.nanpercentile(mag, 95)+0.3, np.nanpercentile(mag, 5)-0.3)\n",