diff --git a/python/lammps.py b/python/lammps.py index 4772a34c8d..99d0b026a5 100644 --- 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