+isi distance matrix with examples

1 files changed, 21 insertions, 0 deletions

diff --git a/examples/isi_matrix.py b/examples/isi_matrix.py
new file mode 100644
index 0000000..a149cd6
--- /dev/null
+++ b/examples/isi_matrix.py
@@ -0,0 +1,21 @@
+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.add_auxiliary_spikes(
+        spk.spike_train_from_string(line), 4000))
+
+print(len(spike_trains))
+
+m = spk.isi_distance_matrix(spike_trains)
+
+plt.imshow(m, interpolation='none')
+plt.show()
+