From 27c28e2caf2234fce94556ad5de68f99160e5a2a Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Wed, 11 Jan 2023 19:15:40 -0500 Subject: [PATCH] accept np.ndarray as index --- dpdata/system.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dpdata/system.py b/dpdata/system.py index 6089db46d..f403b8791 100644 --- a/dpdata/system.py +++ b/dpdata/system.py @@ -339,7 +339,7 @@ def __str__(self): def __getitem__(self, key): """Returns proerty stored in System by key or by idx""" - if isinstance(key, (int, slice, list)): + if isinstance(key, (int, slice, list, np.ndarray)): return self.sub_system(key) return self.data[key]