ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add __getitem__ implementation for NeighList

Browse Source
This commit is contained in:
Richard Berger
2019-09-12 14:14:06 -04:00
parent 34dbcf4f20
commit fec9817086
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/lammps.py
View File

@ -70,6 +70,9 @@ class NeighList:
iatom, numneigh, neighbors = self.lmp.get_neighlist_element_neighbors(self.idx, element) iatom, numneigh, neighbors = self.lmp.get_neighlist_element_neighbors(self.idx, element)
return iatom, numneigh, neighbors return iatom, numneigh, neighbors
def __getitem__(self, element):
return self.get(element)
def __iter__(self): def __iter__(self):
inum = self.size inum = self.size