summaryrefslogtreecommitdiff
path: root/examples/plot.py
blob: da53670a8497baf03bce472e9de56f0d020a9e42 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
""" plot.py

Simple example showing how to load and plot spike trains and their distances.

Copyright 2014, Mario Mulansky <mario.mulansky@gmx.net>

Distributed under the BSD License
"""


from __future__ import print_function

import numpy as np
import matplotlib.pyplot as plt

import pyspike as spk

spike_trains = spk.load_spike_trains_from_txt("PySpike_testdata.txt",
                                              time_interval=(0, 4000))

# 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])
x, y = f.get_plottable_data()

plt.figure()
plt.plot(x, np.abs(y), '--k')

print("Average: %.8f" % f.avrg())
print("Absolute average: %.8f" % f.abs_avrg())


f = spk.spike_distance(spike_trains[0], spike_trains[1])
x, y = f.get_plottable_data()
print(x)
print(y)
#plt.figure()
plt.plot(x, y, '-b')

plt.show()