BUG: GH15426 timezone lost in groupby-agg with cython functions

Author: stephenrauch

doc/source/whatsnew/v0.20.0.txt

@@ -578,6 +578,7 @@ Bug Fixes
 
 - Bug in ``DataFrame.to_stata()`` and ``StataWriter`` which produces incorrectly formatted files to be produced for some locales (:issue:`13856`)
 - Bug in ``pd.concat()`` in which concatting with an empty dataframe with ``join='inner'`` was being improperly handled (:issue:`15328`)
+- Bug in ``groupby.agg()`` incorrectly localizing timezone on ``datetime`` (:issue:`15426`)
 
 
 

pandas/tests/groupby/test_aggregate.py

@@ -6,7 +6,7 @@
 """
 
 from __future__ import print_function
-from datetime import datetime
+from datetime import datetime, timedelta
 from functools import partial
 
 import numpy as np
@@ -738,3 +738,14 @@ def test_agg_over_numpy_arrays(self):
                                 columns=expected_column)
 
         assert_frame_equal(result, expected)
+
+    def test_agg_time_zone_round_trip(self):
+        ts = pd.Timestamp("2016-01-01 12:00:00", tz='US/Pacific')
+        df = pd.DataFrame({'a': 1, 'b': [ts + timedelta(minutes=nn)
+                                         for nn in range(10)]})
+
+        result1 = df.groupby('a')['b'].agg(np.min).iloc[0]
+        result2 = df.groupby('a')['b'].agg(lambda x: np.min(x)).iloc[0]
+
+        self.assertEqual(result1, ts)
+        self.assertEqual(result2, ts)

pandas/types/cast.py

@@ -133,7 +133,8 @@ def trans(x):  # noqa
                 if dtype.tz:
                     # convert to datetime and change timezone
                     from pandas import to_datetime
-                    result = to_datetime(result).tz_localize(dtype.tz)
+                    result = to_datetime(result).tz_localize('utc')
+                    result = result.tz_convert(dtype.tz)
 
     except:
         pass