From 275b4bc473aad3af499fda82f4a04b75bb784210 Mon Sep 17 00:00:00 2001
From: xaver <xaver.roos@t-online.de>
Date: Wed, 8 Jul 2020 14:36:21 +0200
Subject: [PATCH] 08.07 analysis step_stimulus not perfect, but works

---
 second_try.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/second_try.py b/second_try.py
index aaad48a..9d0de30 100644
--- a/second_try.py
+++ b/second_try.py
@@ -40,7 +40,7 @@ for infodataset in infodatasets:
     ID.append(identifier)
 
 
-for dataset in datasets:
+for idx, dataset in enumerate(datasets):
     #input of the function
     frequency, time, amplitude, eodf, deltaf, stimulusf, duration, pause = parse_dataset(dataset)
     dm = np.mean(duration)
@@ -62,7 +62,7 @@ for dataset in datasets:
     freq_all.append(norm.tolist())
     time_all.append(ct_arr)
 
-    plt.plot(ct_arr, norm, color = 'grey', label='fish=%s' % ID)
+    plt.plot(ct_arr, norm, color = 'grey', label='fish=%s' % ID[idx])
 
     # fit function
     sv, sc = curve_fit(step_response, ct_arr[ct_arr < dm], norm[ct_arr < dm]) #step_values and step_cov