Python-сообщество

>>> array = {"H":1,"He" :4}
>>> str = input()
>>> a = array[str]
>>> print(a)

 >>> dct = {'H':1,'He':4}
>>> el = input()
H
>>> print(dct[el])
1
>>> el = input()
He
>>> print(dct[el])
4
>>>

 tabl={"H":1,"He":4} # Здесь перечисляем соответствия символам элемента:атомная масса по прилагаемому образцу, как здесь сделано для двух первых элементов
element1 = input()
try:
    print(tabl[element1])
except KeyError:
    print("Элемент {} отсутствует в таблице Менделеева".format(element1))