From a2fccff7cbe016a5e05d733785b29b4ff1e26f52 Mon Sep 17 00:00:00 2001
From: Kelsey Jordahl <kjordahl@alum.mit.edu>
Date: Tue, 20 Aug 2013 09:06:10 -0400
Subject: [PATCH] BUG: set GeoSeries type in __getitem__ and __getslice__

---
 geopandas/geoseries.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/geopandas/geoseries.py b/geopandas/geoseries.py
index 0a1bea6..8c338c8 100644
--- a/geopandas/geoseries.py
+++ b/geopandas/geoseries.py
@@ -362,6 +362,20 @@ class GeoSeries(Series):
     # Implement pandas methods
     #
 
+    def __getitem__(self, key):
+        val = super(GeoSeries, self).__getitem__(key)
+        if type(val) == Series:
+            val.__class__ = GeoSeries
+            val.crs = self.crs
+        return val
+
+    def __getslice__(self, i, j):
+        val = super(GeoSeries, self).__getslice__(i, j)
+        if type(val) == Series:
+            val.__class__ = GeoSeries
+            val.crs = self.crs
+        return val
+
     @property
     def _can_hold_na(self):
         return False