Trab.Python(Catalogação de flores)(otimizado do otimizado do

import random
escolha = 0
while(escolha!=4):
  print('1-Usuario da informações linha por linha\n2-Usuario  da informações de cada vez\n3-Calibragem\n4-Sair')
  i=int(input('\n'))
  escolha==i
  if(i==1):
    z = []
    x=input('Insira as informações da flor')
    x=x.split(',')
    k=int(input('Valores de k na checagem\n'))
    for i in x:
      a=float(i)
      z.append(a)


  if(i==2):
    z = []
    k=int(input('Valores de k na checagem\n'))
    for i in range(4):
      a = float(input('Insira os dados\n'))
      z.append(a)
  if(i==3):
    check=[]
    arquivo1 = open('iris_test.dat','r')
    for i in arquivo1:
      info = i.split(',')
      c = 0
      z = []
      geralz=[]
      for i in info:
          if c <= 3:
              i = float(i)
              x.append(i)
          if c == 4:
              check.append(i)
          geralz.append(x)
          c += 1

  arquivo = open('iris.dat', 'r')
  geral = []
  ivirgi = []
  ivers = []
  iset = []

  for i in arquivo:
      info = i.split(',')
      c = 0
      x = []
      dist = []
      for i in info:
          if c <= 3:
              i = float(i)
              x.append(i)
              if c == 3:
                  for j in range(4):
                      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 == 4:
              if (i == 'Iris-virginica\n'):
                  ivirgi.append(n2)
              elif (i == 'Iris-setosa\n'):
                  iset.append(n2)
              elif (i == 'Iris-versicolor\n'):
                  ivers.append(n2)
          c += 1
  geral = sorted(geral)
  ivirgi = sorted(ivirgi)
  ivers = sorted(ivers)
  iset = sorted(iset)

  SIvirg = 0
  SIvers = 0
  SIset = 0
  while(k-1 >= len(geral)):
    print('k extrapola o o número de amostras')
    k=int(input('Valores de k na checagem\n'))
  for i in range(k):
      if geral[i] in ivirgi:
          SIvirg += 1
      elif geral[i] in ivers:
          SIvers += 1
      elif geral[i] in iset:
          SIset += 1
  if i==1 or i ==2 :
    check2=[]
    if SIvirg >= SIvers:
        if SIvirg>=SIset:
            check2.append('Os dados da flor inserida, mostram que ela é : Iris-virginica.')
    if SIvers >= SIvirg:
        if SIvers>= SIset:
            check2.append('Os dados da flor inserida, mostram que ela é : Iris-versicolor.')
    if SIset >= SIvirg:
        if SIset>=SIvers:
            check2.append('Os dados da flor inserida, mostram que ela é : Iris-Setosa.')
    print(random.choice(check2))
    arquivo.close()