Programmer un tableau de classification périodique

Bonjour,

Je suis un adepte du langage de programmation Python, et j'aimerai bien m'exercer à écrire des programmes vraiment simple. Je voudrai créer un programme sans fenêtre qui reprend le thème de la classification périodique. Par exemple, dès que je rentre le numéro atomique d'un élément, le programme me retourne l'élément correspondant.
J'ai pensé à faire des
if a==1, print ("H") mais je ne compte pas faire ceci pour tous les éléments du tableau !
Je recherche donc une fonction qui puisse chercher directement l'élément en une syntaxe. Bien sûr il faudra que j'attribue chaque numéro à chaque élément. Mais il y a t il une fonction qui dise :
"if a==1 ou 2 ou 3 ou 4 ou etc, print ("H ou He ou Li ou Be ou etc")

plus simplement ? Je suis à votre écoute.

Sinon il faudra que je fasse une condition pour chaque variable...

Merci d'avance