From 7dd0a099f509154a1f8cd54fa234389b12db6b70 Mon Sep 17 00:00:00 2001
From: David Schaefer <david.schaefer@ufz.de>
Date: Thu, 13 Jun 2019 14:55:23 +0200
Subject: [PATCH] some plotting modifications
---
core.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/core.py b/core.py
index 45b1375f9..566b07c8a 100644
--- a/core.py
+++ b/core.py
@@ -171,17 +171,19 @@ def plot(data, flags, flagmask, varname, flagger, interactive_backend=True, titl
y = data[varname]
flags_ = flags[varname]
nrofflags = len(flagger.flags.categories)
- ax.plot(x, y, '-',markersize=1, color='silver')
+ ax.plot(x, y, '-', markersize=1, color='silver')
if nrofflags == 3:
colors = {0:'silver', 1:'lime', 2:'red'}
- if nrofflags == 4:
+ elif nrofflags == 4:
colors = {0:'silver', 1:'lime', 2:'yellow', 3:'red'}
+ else:
+ warn(f"To many flags.", UserWarning)
+
# plot (all) data in silver
ax.plot(x, y, '-', color='silver', label='data')
# plot (all) missing data in silver
nans = y.isna()
- ylim = plt.ylim()
flagged = flagger.isFlagged(flags_)
idx = y.index[nans & ~flagged]
# ax.vlines(idx, *ylim, linestyles=':', color='silver', label="missing")
--
GitLab