BUG: implement Series.repeat to get expected results, close #1229

Author: wesm

pandas/core/series.py

@@ -609,6 +609,14 @@ def astype(self, dtype):
         casted = com._astype_nansafe(self.values, dtype)
         return self._constructor(casted, index=self.index, name=self.name)
 
+    def repeat(self, reps):
+        """
+        See ndarray.repeat
+        """
+        new_index = self.index.repeat(reps)
+        new_values = self.values.repeat(reps)
+        return Series(new_values, index=new_index, name=self.name)
+
     def reshape(self, newshape, order='C'):
         """
         See numpy.ndarray.reshape

pandas/tests/test_series.py

@@ -2647,6 +2647,19 @@ def test_timeseries_coercion(self):
         self.assert_(isinstance(ser, TimeSeries))
         self.assert_(isinstance(ser.index, DatetimeIndex))
 
+    def test_repeat(self):
+        s = Series(np.random.randn(3), index=['a', 'b', 'c'])
+
+        reps = s.repeat(5)
+        exp = Series(s.values.repeat(5), index=s.index.values.repeat(5))
+        assert_series_equal(reps, exp)
+
+        to_rep = [2, 3, 4]
+        reps = s.repeat(to_rep)
+        exp = Series(s.values.repeat(to_rep),
+                     index=s.index.values.repeat(to_rep))
+        assert_series_equal(reps, exp)
+
 if __name__ == '__main__':
     nose.runmodule(argv=[__file__,'-vvs','-x','--pdb', '--pdb-failure'],
                    exit=False)