From c341dab427f0a0539f9dcf632194f59919d16d56 Mon Sep 17 00:00:00 2001
From: Yannick Ulrich <yannick.ulrich@psi.ch>
Date: Sun, 9 Feb 2020 14:02:59 +0100
Subject: [PATCH] 17: Check input in kplot
---
pymule/plot.py | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/pymule/plot.py b/pymule/plot.py
index 04ac89f..e5cfbf1 100644
--- a/pymule/plot.py
+++ b/pymule/plot.py
@@ -57,6 +57,15 @@ def kplot(sigma, k, sigmar=[], kr=[],
labelk="$K$", labelkr=None,
labelsigma=None, labelsigmar=None):
+ if type(sigma) == np.ndarray:
+ sigma = [sigma]
+ if type(sigmar) == np.ndarray:
+ sigmar = [sigmar]
+ if type(k) == np.ndarray:
+ k = [k]
+ if type(kr) == np.ndarray:
+ kr = [kr]
+
if labelsigma is None:
labelsigma = r"$\D\sigma\,/\,\D " + labelx + "$"
--
GitLab