From 9e154aa1c05117424c3837914d19d646323df367 Mon Sep 17 00:00:00 2001
From: Bert Palm <bert.palm@ufz.de>
Date: Tue, 11 Feb 2020 16:01:36 +0100
Subject: [PATCH] make column slicing, include both bounds
---
dios/dios.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/dios/dios.py b/dios/dios.py
index ba45f24..a748a56 100644
--- a/dios/dios.py
+++ b/dios/dios.py
@@ -8,7 +8,8 @@ from collections import OrderedDict
from pandas.core.dtypes.common import (
is_list_like,
is_iterator,
- is_number
+ is_number,
+ is_integer,
)
from pandas.core.indexing import need_slice
@@ -71,9 +72,9 @@ class _LocIndexer:
stop = keys.index(rslice.stop) if rslice.stop is not None else None
except ValueError:
raise KeyError("The slice start label or the slice stop label is not present in the columns.")
- if not is_number(rslice):
- raise TypeError("The step parameter of the slice must be numeric.")
- return keys[slice(start, stop, rslice.step)]
+ if not is_integer(rslice) and rslice > 0:
+ raise TypeError("The step parameter of the slice must be positive integer.")
+ return keys[slice(start, stop + 1, rslice.step)]
class DictOfSeries:
--
GitLab