Deal with timezone-aware datetime dtypes

GenericMappingTools · Dec 12, 2024 · 44f8106 · 44f8106
1 parent 351247d
commit 44f8106
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pygmt/clib/conversion.py b/pygmt/clib/conversion.py
@@ -193,6 +193,13 @@ def _to_numpy(data: Any) -> np.ndarray:
                 numpy_dtype = np.float64
             data = data.to_numpy(na_value=np.nan)
 
+    # Deal with timezone-aware datetime dtypes.
+    if getattr(dtype, "tz", None):  # pandas.DatetimeTZDtype
+        numpy_dtype = getattr(dtype, "base", None)
+    elif getattr(getattr(dtype, "pyarrow_dtype", None), "tz", None):
+        # pd.ArrayDtype[pa.Timestamp]
+        numpy_dtype = getattr(dtype, "numpy_dtype", None)
+
     array = np.ascontiguousarray(data, dtype=numpy_dtype)
 
     # Check if a np.object_ array can be converted to np.str_.