Skip to content
Snippets Groups Projects
Commit 263899e7 authored by Danilo Ferreira de Lima's avatar Danilo Ferreira de Lima
Browse files

Better plots.

parent 69633e3e
No related branches found
No related tags found
1 merge request!21Calculate SNR and some bug fixes
...@@ -135,7 +135,8 @@ def plot_chi2_intensity(df: pd.DataFrame, filename: str): ...@@ -135,7 +135,8 @@ def plot_chi2_intensity(df: pd.DataFrame, filename: str):
# fill=True, # fill=True,
# ax=ax) # ax=ax)
sns.scatterplot(x=df.chi2_prepca/df.ndof.iloc[0], y=df.xgm_flux_t*1e-3, sns.scatterplot(x=df.chi2_prepca/df.ndof.iloc[0], y=df.xgm_flux_t*1e-3,
s=20, s=30,
marker='+',
#alpha=0.4, #alpha=0.4,
c="tab:red", c="tab:red",
#size=df.root_mean_squared_pca_unc, #size=df.root_mean_squared_pca_unc,
...@@ -542,7 +543,7 @@ if __name__ == '__main__': ...@@ -542,7 +543,7 @@ if __name__ == '__main__':
df_model = pd.read_csv(f'{indir}/model.csv') df_model = pd.read_csv(f'{indir}/model.csv')
df_model.impulse = df_model.impulse.str.replace('i','j').apply(lambda x: complex(x)) df_model.impulse = df_model.impulse.str.replace('i','j').apply(lambda x: complex(x))
df_model.wiener_filter = df_model.wiener_filter.str.replace('i','j').apply(lambda x: complex(x)) df_model.wiener_filter = df_model.wiener_filter.astype(str).str.replace('i','j').apply(lambda x: complex(x))
plot_impulse(df_model, f'impulse.pdf') plot_impulse(df_model, f'impulse.pdf')
plot_wiener(df_model, f'wiener.pdf') plot_wiener(df_model, f'wiener.pdf')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment