from matplotlib import pyplot as plt
import mendeleev
from mendeleev import element
from mendeleev import get_table
import pandas as pd
import pylab

proton=1.00728
netron=1.00876

ptable=get_table('elements')

cols = ['atomic_number', 'symbol','atomic_radius', 'en_pauling', 'block']

isotopes = get_table('isotopes', index_col='id')

merged = pd.merge(ptable[cols], isotopes, how='outer', on='atomic_number')

Mass = []
Div = []

for i in range(406):
    mass=merged['mass'].iloc[i]
    mass_num=merged['mass_number'].iloc[i]
    cnt2=mass_num-merged['atomic_number'].iloc[i]
    cnt1=mass_num-cnt2
    dif=(cnt1*proton+cnt2*netron-mass)*931.5/mass_num
    if (dif>0):
        Mass.append(mass_num)
        Div.append(dif)


pylab.plot(Mass,Div)

pylab.show()







    #ans.iloc[0]
    #print(merged[merged['atomic_number']==i])
    #print(ans['symbol'])
#kek=merged[merged['atomic_number']==28]
#print(kek[kek['spin']==1.5])