python - 将日期添加到 float 的 DataArray(或反之亦然)

Question

我对python有点陌生。我有一个 xarray DataArray 包含一个由浮点数组成的变量。这些浮点数应该是秒。我想将这些浮点数添加到给定日期:

2021-01-01 00:00:00 + 1 =  2021-01-01 00:00:01
...
2021-01-01 00:00:00 + 120 = 2021-01-01 00:02:00

为了有一个我可以使用的日期时间数组(例如提取给定时间间隔之间的事件)
我尝试使用 timedelta 但它不喜欢 DataArray。有解决方法吗？

Answer 1

对于标准日历操作，xarray works best with NumPy datetime64 and timedelta64 types .在您的特定情况下，我建议您将浮点数的 DataArray 转换为 timedelta64 (考虑到秒的单位)，并将这些值添加到 datetime64实例。这将产生一个 datetime64[ns] 类型的 DataArray ，您可以在其他操作中使用它:

In [1]: import numpy as np; import xarray as xr

In [2]: data = np.arange(10.)

In [3]: da = xr.DataArray(data, dims=["x"])

In [4]: np.datetime64("2000-01-01") + da.astype("timedelta64[s]")
Out[4]:
<xarray.DataArray (x: 10)>
array(['2000-01-01T00:00:00.000000000', '2000-01-01T00:00:01.000000000',
       '2000-01-01T00:00:02.000000000', '2000-01-01T00:00:03.000000000',
       '2000-01-01T00:00:04.000000000', '2000-01-01T00:00:05.000000000',
       '2000-01-01T00:00:06.000000000', '2000-01-01T00:00:07.000000000',
       '2000-01-01T00:00:08.000000000', '2000-01-01T00:00:09.000000000'],
      dtype='datetime64[ns]')
Dimensions without coordinates: x