From 7989b2d337a0e5d2e0223d7fdec73833ff47c7bb Mon Sep 17 00:00:00 2001 From: Mario Mulansky Date: Tue, 3 Feb 2015 16:40:55 +0100 Subject: first version of psth profile --- pyspike/psth.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 pyspike/psth.py (limited to 'pyspike/psth.py') diff --git a/pyspike/psth.py b/pyspike/psth.py new file mode 100644 index 0000000..8516460 --- /dev/null +++ b/pyspike/psth.py @@ -0,0 +1,27 @@ +""" + +Module containing functions to compute the PSTH profile + +Copyright 2015, Mario Mulansky + +Distributed under the BSD License +""" + +import numpy as np +from pyspike import PieceWiseConstFunc + + +# Computes the Peristimulus time histogram of a set of spike trains +def psth(spike_trains, bin_size): + + bins = int((spike_trains[0][-1] - spike_trains[0][0]) / bin_size) + + N = len(spike_trains) + combined_spike_train = spike_trains[0][1:-1] + for i in xrange(1, len(spike_trains)): + combined_spike_train = np.append(combined_spike_train, + spike_trains[i][1:-1]) + + vals, edges = np.histogram(combined_spike_train, bins, density=False) + bin_size = edges[1]-edges[0] + return PieceWiseConstFunc(edges, vals/(N*bin_size)) -- cgit v1.2.3