BUG: Pandas can't restore index from parquet with offset-specified timezone #35997

Author: alippai

pandas/_libs/tslibs/timezones.pyx

@@ -15,7 +15,7 @@ from pytz.tzinfo import BaseTzInfo as _pytz_BaseTzInfo
 
 UTC = pytz.utc
 
-
+import re
 import numpy as np
 
 cimport numpy as cnp
@@ -103,7 +103,14 @@ cpdef inline tzinfo maybe_get_tz(object tz):
             if isinstance(tz, _dateutil_tzfile) and '.tar.gz' in tz._filename:
                 tz._filename = zone
         else:
-            tz = pytz.timezone(tz)
+            if tz.startswith('+') or tz.startswith('-') or tz.startswith('UTC+') or tz.startswith('UTC-'):
+                m = re.match(r'(?P<sign>[+-])(?P<hour>\d{2}):(?P<minutes>\d{2})', tz).groupdict()
+                if m['sign'] == '-':
+                    tz = pytz.FixedOffset(-60 * int(m['hour']) + int(m['minutes']))
+                else:
+                    tz = pytz.FixedOffset(60*int(m['hour']) + int(m['minutes']))
+            else:
+                tz = pytz.timezone(tz)
     elif is_integer_object(tz):
         tz = pytz.FixedOffset(tz / 60)
     elif isinstance(tz, tzinfo):

pandas/tests/io/test_parquet.py

@@ -712,6 +712,12 @@ def test_timestamp_nanoseconds(self, pa):
         df = pd.DataFrame({"a": pd.date_range("2017-01-01", freq="1n", periods=10)})
         check_round_trip(df, pa, write_kwargs={"version": "2.0"})
 
+    @td.skip_if_no("pyarrow", min_version="0.14")
+    def test_timezone_aware_index(self, pa):
+        idx = [datetime.datetime.now(datetime.timezone.utc)]
+        df = pd.DataFrame(index=idx)
+        check_round_trip(df, pa, write_kwargs={"version": "2.0"})
+
     @td.skip_if_no("pyarrow", min_version="0.17")
     def test_filter_row_groups(self, pa):
         # https://github.com/pandas-dev/pandas/issues/26551
@@ -850,3 +856,12 @@ def test_empty_dataframe(self, fp):
         expected = df.copy()
         expected.index.name = "index"
         check_round_trip(df, fp, expected=expected)
+
+    def test_timezone_aware_index(self, fp):
+        idx = [
+            datetime.datetime.strptime(
+                "2019-01-04T16:41:24+0200", "%Y-%m-%dT%H:%M:%S%z"
+            )
+        ]
+        df = pd.DataFrame(index=idx)
+        check_round_trip(df, fp)