diff options
author | Mario Mulansky <mario.mulansky@gmx.net> | 2014-10-22 22:17:27 +0200 |
---|---|---|
committer | Mario Mulansky <mario.mulansky@gmx.net> | 2014-10-22 22:17:27 +0200 |
commit | 11f05d9dac3711d89db37a043db3d9437958c6f3 (patch) | |
tree | ddcad0d56d33ff69907a92fe4c5cbcb734311c27 /pyspike/function.py | |
parent | 7dbb5f87321fc89b6c84552cfd821f96fe364bbe (diff) |
avrg functions now take intervals
Diffstat (limited to 'pyspike/function.py')
-rw-r--r-- | pyspike/function.py | 51 |
1 files changed, 44 insertions, 7 deletions
diff --git a/pyspike/function.py b/pyspike/function.py index 14ad7bd..e340096 100644 --- a/pyspike/function.py +++ b/pyspike/function.py @@ -83,20 +83,24 @@ class PieceWiseConstFunc(object): """ if interval is None: # no interval given, average over the whole spike train - return np.sum((self.x[1:]-self.x[:-1]) * self.y) / \ + a = np.sum((self.x[1:]-self.x[:-1]) * self.y) / \ (self.x[-1]-self.x[0]) else: + # find the indices corresponding to the interval start_ind = np.searchsorted(self.x, interval[0], side='right') end_ind = np.searchsorted(self.x, interval[1], side='left')-1 - print(start_ind, end_ind) assert start_ind > 0 and end_ind < len(self.x), \ "Invalid averaging interval" - a = np.sum((self.x[start_ind+1:end_ind] - - self.x[start_ind:end_ind-1]) * + # first the contribution from between the indices + a = np.sum((self.x[start_ind+1:end_ind+1] - + self.x[start_ind:end_ind]) * self.y[start_ind:end_ind]) - a += (self.x[start_ind]-interval[0]) * self.y[start_ind] + # correction from start to first index + a += (self.x[start_ind]-interval[0]) * self.y[start_ind-1] + # correction from last index to end a += (interval[1]-self.x[end_ind]) * self.y[end_ind] - return a / (interval[1]-interval[0]) + a /= (interval[1]-interval[0]) + return a def add(self, f): """ Adds another PieceWiseConst function to this function. @@ -202,10 +206,43 @@ class PieceWiseLinFunc: :returns: the average a. :rtype: double """ + + def intermediate_value(x0, x1, y0, y1, x): + """ computes the intermediate value of a linear function """ + return y0 + (y1-y0)*(x-x0)/(x1-x0) + if interval is None: # no interval given, average over the whole spike train - return np.sum((self.x[1:]-self.x[:-1]) * 0.5*(self.y1+self.y2)) / \ + a = np.sum((self.x[1:]-self.x[:-1]) * 0.5*(self.y1+self.y2)) / \ (self.x[-1]-self.x[0]) + else: + # find the indices corresponding to the interval + start_ind = np.searchsorted(self.x, interval[0], side='right') + end_ind = np.searchsorted(self.x, interval[1], side='left')-1 + assert start_ind > 0 and end_ind < len(self.x), \ + "Invalid averaging interval" + # first the contribution from between the indices + a = np.sum((self.x[start_ind+1:end_ind+1] - + self.x[start_ind:end_ind]) * + 0.5*(self.y1[start_ind:end_ind] + + self.y2[start_ind:end_ind])) + # correction from start to first index + a += (self.x[start_ind]-interval[0]) * 0.5 * \ + (self.y2[start_ind-1] + + intermediate_value(self.x[start_ind-1], self.x[start_ind], + self.y1[start_ind-1], + self.y2[start_ind-1], + interval[0] + )) + # correction from last index to end + a += (interval[1]-self.x[end_ind]) * 0.5 * \ + (self.y1[end_ind] + + intermediate_value(self.x[end_ind], self.x[end_ind+1], + self.y1[end_ind], self.y2[end_ind], + interval[1] + )) + a /= (interval[1]-interval[0]) + return a def add(self, f): """ Adds another PieceWiseLin function to this function. |