pint-xarray icon indicating copy to clipboard operation
pint-xarray copied to clipboard

interaction with `timedelta64`

Open keewis opened this issue 1 year ago • 2 comments

Arrays with dtype timedelta64 are basically quantities with dimension [time], and they even have explicit units.

It would be really useful to be able to convert between those and pint quantities. The core logic should probably live in pint, but I think we should figure out a way to make the conversion easier:

arr = xr.Dataset(
    coords={
        "delta1": ("time", np.arange(100, dtype="timedelta64[s]")),
        "delta2": ("time", np.arange(100), {"units": "s"})
    }
).pint.quantify()
arr.pint.to({"delta1": "s", "delta2": "timedelta64[s]"})

(not sure if overloading .to would be worth it, but if not I'm sure we can come up with something better).

Similarly, I'd like to be able to infer the frequency of a datetime64 object and get the frequency (if any) as a quantity. It seems this works (found here):

from pandas.tseries.frequencies import to_offset

timedelta = to_offset(xr.infer_freq(time)).delta.to_numpy()

where the timedelta would then be converted to pint using the functionality described above.

keewis avatar Jun 14 '23 09:06 keewis

I had the same question and posted it here on stackoverflow.

lumbric avatar May 06 '24 12:05 lumbric

The core logic should probably live in pint

Here's a PR for the pint logic https://github.com/hgrecco/pint/pull/1978

andrewgsavage avatar May 06 '24 22:05 andrewgsavage