diff --git a/xarray/indexes/range_index.py b/xarray/indexes/range_index.py
index 2b9a5e5071a..34e6c2f7c00 100644
--- a/xarray/indexes/range_index.py
+++ b/xarray/indexes/range_index.py
@@ -9,7 +9,7 @@
from xarray.core.coordinate_transform import CoordinateTransform
from xarray.core.dataarray import DataArray
from xarray.core.indexes import CoordinateTransformIndex, Index, PandasIndex
-from xarray.core.indexing import IndexSelResult, normalize_slice
+from xarray.core.indexing import IndexSelResult
from xarray.core.variable import Variable
@@ -79,14 +79,10 @@ def equals(
)
def slice(self, sl: slice) -> "RangeCoordinateTransform":
- sl = normalize_slice(sl, self.size)
-
- # TODO: support reverse transform (i.e., start > stop)?
- assert sl.start < sl.stop
-
- new_size = (sl.stop - sl.start) // sl.step
- new_start = self.start + sl.start * self.step
- new_stop = new_start + new_size * sl.step * self.step
+ new_range = range(self.size)[sl]
+ new_size = len(new_range)
+ new_start = self.start + new_range.start * self.step
+ new_stop = self.start + new_range.stop * self.step
return type(self)(
new_start, new_stop, new_size, self.coord_name, self.dim, dtype=self.dtype
diff --git a/xarray/tests/test_range_index.py b/xarray/tests/test_range_index.py
index 05bf4b3172e..d0644ba73a2 100644
--- a/xarray/tests/test_range_index.py
+++ b/xarray/tests/test_range_index.py
@@ -121,6 +121,28 @@ def test_range_index_isel() -> None:
expected = create_dataset_arange(0.0, 1.0, 0.2)
assert_identical(actual, expected, check_default_indexes=False)
+ actual = ds.isel(x=slice(None, None, -1))
+ expected = create_dataset_arange(0.9, -0.1, -0.1)
+ assert_identical(actual, expected, check_default_indexes=False)
+
+ actual = ds.isel(x=slice(None, 4, -1))
+ expected = create_dataset_arange(0.9, 0.4, -0.1)
+ assert_identical(actual, expected, check_default_indexes=False)
+
+ actual = ds.isel(x=slice(8, 4, -1))
+ expected = create_dataset_arange(0.8, 0.4, -0.1)
+ assert_identical(actual, expected, check_default_indexes=False)
+
+ actual = ds.isel(x=slice(8, None, -1))
+ expected = create_dataset_arange(0.8, -0.1, -0.1)
+ assert_identical(actual, expected, check_default_indexes=False)
+
+ # https://github.com/pydata/xarray/issues/10441
+ ds2 = create_dataset_arange(0.0, 3.0, 0.1)
+ actual = ds2.isel(x=slice(4, None, 3))
+ expected = create_dataset_arange(0.4, 3.0, 0.3)
+ assert_identical(actual, expected, check_default_indexes=False)
+
# scalar
actual = ds.isel(x=0)
expected = xr.Dataset(coords={"x": 0.0})
pFad - Phonifier reborn
Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.
Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.
Alternative Proxies:
Alternative Proxy
pFad Proxy
pFad v3 Proxy
pFad v4 Proxy