#1. feladat
fbe=open('atomok.txt','r')
sorok=fbe.readlines()
lista0=[]
for elem in sorok:
    lista0.append(elem.strip())
i=0
while i<len(lista0):
    ujsor=''
    pont=len(lista0[i])
    psz=0
    j=0
    while j<len(lista0[i]):
        if lista0[i][j]==',':
            ujsor+='.'
            pont=j
            psz+=1
        elif pont<j and lista0[i][j]==' ':
            ujsor+=''
        elif lista0[i][j]=='(':
            while lista0[i][j]!=' ' and j<len(lista0[i])-1:
                ujsor+=''
                j+=1
            if lista0[i][j]==' ':
                ujsor+=' '
            elif psz==1:
                ujsor+=' 0.0'
            else:
                ujsor+=''
        else:
            ujsor+=lista0[i][j]
        j+=1
    lista0[i]=ujsor
    i+=1
lista=[]
for elem in lista0:
    lista.append(elem.split())
#2. feladat
print('2. feladat:')
v=input('Kérem egy elem vegyjelét: ')
i=0
while i<len(lista):
    if lista[i][1]==v:
        print('Z={0} A={1} m={2} ea={3}'.format(lista[i][0],lista[i][2],lista[i][3],lista[i][4]))
    i+=1
#3. feladat
print('\n3. feladat')
izotopszam=[]
elemek=[]
maxert=0
isz=0
while isz<int(lista[len(lista)-1][0]):
    i=0
    idb=0
    while i<len(lista):
        if lista[i][1] not in elemek:
            elemek.append(lista[i][1])
        if int(lista[i][0])==len(izotopszam)+1:
            idb+=1
        i+=1
    izotopszam.append(idb)
    if maxert<idb:
        maxert=idb
    isz+=1
print('A legnagyobb izotópszám: {0}'.format(maxert))
print('Az elemek: ',end='')
maxi=[]
i=0
while i<len(izotopszam):
    if izotopszam[i]==maxert:
        maxi.append(i)
    i+=1
kiir=[]
k=0
while k<len(maxi):
    j=0
    while j<len(lista):
        if int(lista[j][0])==maxi[k]+1 and lista[j][1] not in kiir:
            kiir.append(lista[j][1])
            if k==0:
                print(lista[j][1],end='')
            else:
                print(',',lista[j][1],end='')
        j+=1
    k+=1
print()
#4. feladat
def tomeg(vj):
    i=0
    izotoposszeg=eaosszeg=0
    idb=0
    while i<len(lista):
        if lista[i][1]==vj:
            izotoposszeg+=float(lista[i][3])*float(lista[i][4])
            eaosszeg+=float(lista[i][4])
            idb+=1
        i+=1
    if eaosszeg!=0:
        relatomtomeg=izotoposszeg/eaosszeg
    else:
        relatomtomeg=0
    return relatomtomeg
#5. feladat
fki=open('tomegek.txt','w')
i=0
while i<len(elemek):
    if tomeg(elemek[i])!=0:
        print(elemek[i],tomeg(elemek[i]),file=fki)
    i+=1
fki.close()
