Update main_cali.py

wang1hm · Mar 12, 2023 · 2cc83b7 · 2cc83b7
1 parent 43fd94c
commit 2cc83b7
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/lamostabs/main_cali.py b/lamostabs/main_cali.py
@@ -24,12 +24,16 @@
     data.iloc[:,i]=pd.to_numeric(data.iloc[:,i])
 
 #pd.to_numeric(data.iloc[:,5])
-mag = data.iloc[0][5:9]
-mag_err = data.iloc[0][9:13]
-#p = data['redshift'][0]
-z = data['redshift'][0]
-photometry = data['Photometry'][0]
-name = data['basename'][0]
+spec_index=0
+mag = data.iloc[spec_index][5:9]
+mag_err = data.iloc[spec_index][9:13]
+#p = data['redshift'][spec_index]
+z = data['redshift'][spec_index]
+photometry = data['Photometry'][spec_index]
+name = data['basename'][spec_index]
 Res,arr_f1 = CaliFlux.RedBlue(mag,mag_err,z,photometry,str(name))
+
+
 plt.plot(Res[2],Res[0],zorder = 0)
-plt.scatter([4686.,6165.,7481.],arr_f1,c='r',zorder=1)
+plt.scatter([4686.,6165.,7481.],arr_f1,c='r',zorder=1)
+