Is it possible to selecting dataset by time range when range is different for every pixel in pythons xarray module

Question

I try to select only this part of the data within a specific time range that is different for every pixel.

For indexing, I have two np.datetime64[ns] xr.DataArrays with shape(lat:152, lon:131) named time_range_min, time_range_max One is holding the start dates and the other one the end dates.

I try this for selecting the data

dataset = data.sel(time=slice(time_range_min, time_range_max))

but it yields

cannot use non-scalar arrays in a slice for xarray indexing: <xarray.DataArray 'NDVI' (lat: 152, lon: 131)>

If I cannot use non-scalar arrays it means that it is in general not possible to do this, or can I transform my arrays?

Answer 1

If "time" is a list of dates in string that is ordered from past to present (e.g. ["10-20-2021", "10-21-2021", ...]:

import numpy as np
listOfMinMaxTimeRanges = [time_range_min, time_range_max]
specifiedRangeOfTimeIndexedList = []
for indexingListOfMinMaxTimeRanges in range(len(listOfMinMaxTimeRanges)):
    specifiedRangeOfTimeIndexed = [specifiedRangeOfTime for specifiedRangeOfTime in np.arange(0, len(time), 1) if time.index(listOfMinMaxTimeRanges[0][indexingListOfMinMaxTimeRanges]) <= specifiedRangeOfTime <= time.index(listOfMinMaxTimeRanges[1][indexingListOfMinMaxTimeRanges])]
    for indexes in specifiedRangeOfTimeIndexed:
        specifiedRangeOfTimeIndexedList.append(indexes)

Depending on how your dataset is structured:

dataset = data.sel(time = specifiedRangeOfTimeIndexedList)

or

dataset = data.sel(time = time[specifiedRangeOfTimeIndexedList])

or

dataset = dataset[time[specifiedRangeOfTimeIndexedList]]

or

dataset = dataset[:, time[specifiedRangeOfTimeIndexedList]]

or

dataset = dataset[time[specifiedRangeOfTimeIndexedList], :, :]

or

dataset = dataset[specifiedRangeOfTimeIndexedList]

...