diff options
Diffstat (limited to 'examples/test_data.py')
-rw-r--r-- | examples/test_data.py | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/examples/test_data.py b/examples/test_data.py new file mode 100644 index 0000000..bddcb15 --- /dev/null +++ b/examples/test_data.py @@ -0,0 +1,28 @@ +# compute the isi distance of some test data +from __future__ import print_function + +import numpy as np +import matplotlib.pyplot as plt + +import pyspike as spk + +# first load the data +spike_trains = [] +spike_file = open("SPIKY_testdata.txt", 'r') +for line in spike_file: + spike_trains.append(spk.spike_train_from_string(line)) + +# plot the spike time +for (i,spikes) in enumerate(spike_trains): + plt.plot(spikes, i*np.ones_like(spikes), 'o') + +f = spk.isi_distance(spike_trains[0], spike_trains[1], 4000) +x, y = f.get_plottable_data() + +plt.figure() +plt.plot(x, y, '-k') + +print("Average: %.8f" % f.avrg()) +print("Absolute average: %.8f" % f.abs_avrg()) + +plt.show() |