Как упростить код?

Question

[qetuo3]

****input.txt***
5
7
******************

fin = open('input.txt',)
fout = open('output.txt','w')

a = fin.readline(3).split()
A = a[0]

b = fin.readline(3).split()
B = b[0]

if A < B:
	fout.write('<')
elif A > B:
	fout.write('>')
elif A == B:
	fout.write('=')

fin.close()
fout.close()

если верить сайту acmp.ru, то этот код не успевает сработать за 1 секунду

Comments

[dadasay]

Тут дело не во времени, а в коде. В нем ошибка

Answer 1

[dadasay]

a, b = int(input()), int(input()) 
if a > b:
	print('>')
elif a < b:
	print('<')
elif a == b:
	print('=')  


#Переработанный ваш
fin_name = 'input.txt'
fout_name = "output.txt"

with open(fin_name) as fin:
    a, b = map(lambda x: int(x.strip()), fin.readlines())

result = None
if a > b:
    result = '>'
elif a < b:
    result = '<'
elif a == b:
    result = '='

with open(fout_name, 'w') as fout:
    fout.write(result)

Comments

[qetuo3]

мне нужно присвоить числа из input.txt в переменные, а не ввод пользователем и выводить в output.txt, а не в консоль.

[dadasay]

qetuo3, Изменил код

Answer 2

[Сергей Чамкин]

код не успевает сработать за 1 секунду

Разве?? У вас ошибка во время исполнения!