add from_Z method

Author: haidi-ustc

dpdata/periodic_table.py

@@ -3,11 +3,16 @@
 
 fpdt=str(Path(__file__).absolute().parent / "periodic_table.json")
 _pdt=loadfn(fpdt)
+ELEMENTS=['H', 'He', 'Li', 'Be', 'B', 'C', 'N', 'O', 'F', 'Ne', 'Na', 'Mg', 'Al', 'Si', 'P', 'S', 'Cl', 'Ar', 'K', 'Ca', 'Sc', 'Ti', 'V', 'Cr', \
+         'Mn', 'Fe', 'Co', 'Ni', 'Cu', 'Zn', 'Ga', 'Ge', 'As', 'Se', 'Br', 'Kr', 'Rb', 'Sr', 'Y', 'Zr', 'Nb', 'Mo', 'Tc', 'Ru', 'Rh', 'Pd', 'Ag',\
+         'Cd', 'In', 'Sn', 'Sb', 'Te', 'I', 'Xe', 'Cs', 'Ba', 'La', 'Ce', 'Pr', 'Nd', 'Pm', 'Sm', 'Eu', 'Gd', 'Tb', 'Dy', 'Ho', 'Er', 'Tm', 'Yb',\
+         'Lu', 'Hf', 'Ta', 'W', 'Re', 'Os', 'Ir', 'Pt', 'Au', 'Hg', 'Tl', 'Pb', 'Bi', 'Po', 'At', 'Rn', 'Fr', 'Ra', 'Ac', 'Th', 'Pa', 'U', 'Np', \
+         'Pu', 'Am', 'Cm', 'Bk', 'Cf', 'Es', 'Fm', 'Md', 'No', 'Lr']
 
 class Element:
 
     def __init__(self, symbol: str):
-        assert symbol in _pdt.keys()
+        assert symbol in ELEMENTS
         self.symbol = "%s" % symbol
         d = _pdt[symbol]
         self._Z = d['atomic_no']
@@ -22,6 +27,12 @@ def __str__(self):
 
     def __repr__(self):
         return "Element : %s"%self.symbol
+    
+    @classmethod
+    def from_Z(cls,Z):
+        assert(Z>0)
+        assert(Z<len(ELEMENTS))
+        return cls(ELEMENTS[Z-1])
 
     @property
     def Z(self):

tests/test_periodic_table.py

@@ -21,6 +21,7 @@ def test_H(self):
         self.assertEqual(H.mass,data['atomic_mass'])
         self.assertEqual(H.radius,data['radius'])
         self.assertEqual(H.calculated_radius,data['calculated_radius'])
+        self.assertEqual(H.X,dpdata.periodic_table.Element.from_Z(1).X)
 
 if __name__ == '__main__':
     unittest.main()