Trab.Python(3.1)

import random
z = []
geral = []
nomes = []
k=int(input('Valores de k na checagem'))

for i in range(6):###Inseri os valores do usuario
    a = float(input('Insira os dados'))
    z.append(a)
arquivo = open('coluna_calibrar.dat', 'r')###abre o arquivo
DH = []
SL = []
NO = []

for i in arquivo:
    info = i.split(' ')
    c = 0
    x = []
    dist = []
    for i in info:
        if c <= 5:
            i = float(i)
            x.append(i)
            if c == 5:
                for j in range(6):
                    n=(z[j]-x[j])**2
                    dist.append(n)
                n1=0
                n2=0
                for x in dist:
                    n1+=x
                n2=n1**(1/2)
                geral.append(n2)
        if c == 6:
            if (i == 'DH\n'):
                DH.append(n2)
            elif (i == 'SL\n'):
                SL.append(n2)
            elif (i == 'NO\n'):
                NO.append(n2)
        c += 1
geral = sorted(geral)
ivirgi = sorted(DH)
ivers = sorted(SL)
iset = sorted(NO)

check = []
for i in range(k):
    check.append(geral[i])
for i in check:
    if i in DH:
        nomes.append('DH')
    elif i in SL:
        nomes.append('SL')
    elif i in NO:
        nomes.append('NO')
a1 = 0
a2 = 0
a3 = 0

for i in nomes:
    if i == 'DH':
        a1 += 1
    elif i == 'SL':
        a2 += 1
    elif i == 'NO':
        a3 += 1
'''print()
print(check)
print()
print(geral)
print()
print(DH)
print()
print(SL)
print()
print(NO)
print()
print(nomes)
print()'''
check2=[]
if a1 >= a2:
    if a1>=a3:
        check2.append('Os dados da flor inserida mostram que ela é : Hernia de Disco')
if a2 >= a1:
    if a2>= a3:
        check2.append('Os dados da flor inserida mostram que ela é : Espondilolistese.')
if a3 >= a1:
    if a3>=a2:
        check2.append('Os dados da flor inserida mostram que ela é : Normal')
print(random.choice(check2))
arquivo.close()