From 263899e72921a2397fa101d7d4c209799e70b27a Mon Sep 17 00:00:00 2001
From: Danilo Ferreira de Lima <danilo.enoque.ferreira.de.lima@xfel.de>
Date: Tue, 7 Nov 2023 15:24:23 +0100
Subject: [PATCH] Better plots.
---
pes_to_spec/test/prepare_plots.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/pes_to_spec/test/prepare_plots.py b/pes_to_spec/test/prepare_plots.py
index a0b4222..73c8052 100755
--- a/pes_to_spec/test/prepare_plots.py
+++ b/pes_to_spec/test/prepare_plots.py
@@ -135,7 +135,8 @@ def plot_chi2_intensity(df: pd.DataFrame, filename: str):
# fill=True,
# ax=ax)
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,
c="tab:red",
#size=df.root_mean_squared_pca_unc,
@@ -542,7 +543,7 @@ if __name__ == '__main__':
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.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_wiener(df_model, f'wiener.pdf')
--
GitLab