From d971b61f801cf6732b0ef602dc2235161c83a52b Mon Sep 17 00:00:00 2001
From: David Rotermund <54365609+davrot@users.noreply.github.com>
Date: Mon, 10 Jul 2023 18:46:54 +0200
Subject: [PATCH] Add files via upload

---
 test_average_calculate.py |  2 +-
 test_average_show.py      | 16 ++++++++++------
 2 files changed, 11 insertions(+), 7 deletions(-)

diff --git a/test_average_calculate.py b/test_average_calculate.py
index 7e6d9b7..6b67f0b 100644
--- a/test_average_calculate.py
+++ b/test_average_calculate.py
@@ -15,7 +15,7 @@ start_position: int = 0
 start_position_coefficients: int = 100
 remove_heartbeat: bool = True  # i.e. use SVD
 bin_size: int = 4
-threshold: float | None = 0.1  # Between 0 and 1.0
+threshold: float | None = 0.05  # Between 0 and 1.0
 
 
 display_logging_messages: bool = False
diff --git a/test_average_show.py b/test_average_show.py
index a8cb419..b2d8fca 100644
--- a/test_average_show.py
+++ b/test_average_show.py
@@ -10,9 +10,15 @@ example_position_y: int = 440
 bin_size: int = 4
 bin_size_post: int | None = None
 
-data = np.load("result.npz")
-result = data["result"]
-mask = data["mask"]
+result = np.load("result.npy")
+mask = np.load("mask.npy")
+count_not_nan = np.load("count_not_nan.npy")
+
+# plt.imshow(count_not_nan.mean(axis=0))
+# plt.colorbar()
+# plt.title("Not nan")
+# plt.show()
+# exit()
 
 example_position_x = example_position_x // bin_size
 example_position_y = example_position_y // bin_size
@@ -26,6 +32,4 @@ if show_example_timeseries:
 
 if play_movie:
     ani = Anime()
-    ani.show(
-        result - 1.0, mask=mask, vmin_scale=0.5, vmax_scale=0.5
-    )  # , vmin=0.98)  # , vmin=1.0, vmax_scale=1.0)
+    ani.show(result, mask=mask, vmin_scale=0.01, vmax_scale=0.01)