Commit fec98170 authored by Richard Berger's avatar Richard Berger
Browse files

Add __getitem__ implementation for NeighList

parent 34dbcf4f

python/lammps.py

+3 −0
Original line number Diff line number Diff line
@@ -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

CRA Git | Maintained and supported by SUSTech CRA and CCSE