Add __getitem__ implementation for NeighList

2019-09-12 14:14:06 -04:00
parent 34dbcf4f20
commit fec9817086
1 changed files with 3 additions and 0 deletions
--- a/python/lammps.py
+++ b/python/lammps.py
@ -70,6 +70,9 @@ class NeighList:
        iatom, numneigh, neighbors = self.lmp.get_neighlist_element_neighbors(self.idx, element)
        return iatom, numneigh, neighbors

+    def __getitem__(self, element):
+        return self.get(element)
+
    def __iter__(self):
        inum = self.size