Merge pull request #20 from mariomulansky/develop

Develop merge for version 0.4

1 files changed, 15 insertions, 0 deletions

diff --git a/pyspike/SpikeTrain.py b/pyspike/SpikeTrain.py
index 9127b60..4b59a5d 100644
--- a/pyspike/SpikeTrain.py
+++ b/pyspike/SpikeTrain.py
@@ -32,6 +32,21 @@ class SpikeTrain(object):
             self.t_start = 0.0
             self.t_end = float(edges)
 
+    def __getitem__(self, index):
+        """ Returns the time of the spike given by index.
+
+        :param index: Index of the spike.
+        :return: spike time.
+        """
+        return self.spikes[index]
+
+    def __len__(self):
+        """ Returns the number of spikes.
+        
+        :return: Number of spikes.
+        """
+        return len(self.spikes)
+
     def sort(self):
         """ Sorts the spike times of this spike train using `np.sort`
         """