@@ -699,6 +699,32 @@ def test_resample_consistency(self):
699
699
assert_series_equal (s10_2 , r10_2 )
700
700
assert_series_equal (s10_2 , rl )
701
701
702
+ def test_resample_timegrouper (self ):
703
+ # GH 7227
704
+ dates = [datetime (2014 , 10 , 1 ), datetime (2014 , 9 , 3 ),
705
+ datetime (2014 , 11 , 5 ), datetime (2014 , 9 , 5 ),
706
+ datetime (2014 , 10 , 8 ), datetime (2014 , 7 , 15 )]
707
+
708
+ df = DataFrame (dict (A = dates , B = np .arange (len (dates ))))
709
+ result = df .set_index ('A' ).resample ('M' , how = 'count' )
710
+ exp_idx = pd .DatetimeIndex (['2014-07-31' , '2014-08-31' , '2014-09-30' ,
711
+ '2014-10-31' , '2014-11-30' ], freq = 'M' , name = 'A' )
712
+ expected = DataFrame ({'B' : [1 , 0 , 2 , 2 , 1 ]}, index = exp_idx )
713
+ assert_frame_equal (result , expected )
714
+
715
+ result = df .groupby (pd .Grouper (freq = 'M' , key = 'A' )).count ()
716
+ assert_frame_equal (result , expected )
717
+
718
+ df = DataFrame (dict (A = dates , B = np .arange (len (dates )), C = np .arange (len (dates ))))
719
+ result = df .set_index ('A' ).resample ('M' , how = 'count' )
720
+ expected = DataFrame ({'B' : [1 , 0 , 2 , 2 , 1 ], 'C' : [1 , 0 , 2 , 2 , 1 ]},
721
+ index = exp_idx , columns = ['B' , 'C' ])
722
+ assert_frame_equal (result , expected )
723
+
724
+ result = df .groupby (pd .Grouper (freq = 'M' , key = 'A' )).count ()
725
+ assert_frame_equal (result , expected )
726
+
727
+
702
728
def _simple_ts (start , end , freq = 'D' ):
703
729
rng = date_range (start , end , freq = freq )
704
730
return Series (np .random .randn (len (rng )), index = rng )
0 commit comments