Печать 2 строк под пользовательским вводом?

Question

Я пробовал распечатать 2 строки под кодом, который пользователь вводит из текстового файла.

мой текстовый файл:

86947367 
banana 
5 

78364721 
apple 
3 

35619833 
orange 
2 

84716491 
sweets 
8 

46389121 
chicken 
10 

Таким образом, если пользователь вводит, например, 46389121 то он будет печатать

chicken 
    10 

Я пробовал:

file = 'read_it.txt' 
user = input('Enter code') 

with open(file, 'r') as f: 
    lines = f.readlines() 
    for i, line in enumerate(lines): # enumerate will count and keep track  of the lines 
     if line == user: 
      print("{}\n{}".format(lines[i+1], lines[i+2])) 

, но это дает мне выход пустых строк, любая помощь, пожалуйста?

Answer 1

Попробуйте зачистки линии перед сравнением с использованием rstrip():

file = 'read_it.txt' 
user = input('Enter code') 

with open(file, 'r') as f: 
    lines = f.readlines() 
    for i, line in enumerate(lines): # enumerate will count and keep track  of the lines 
     if line.rstrip() == user: 
      print("{}\n{}".format(lines[i+1], lines[i+2])) 

Имейте в виду, что вместо загрузки всех файлов одновременно, вы можете просто перебирать над ним, как так:

file = 'read_it.txt' 
user = input('Enter code') 

with open(file, 'r') as f: 
    for line in f: 
     if line.rstrip() == user: 
      print(next(f).rstrip()) 
      print(next(f).rstrip()) 
      break 

---

Функция rstrip() удаляет пробелы в конце строки. Пробелы включают в себя , \n, \t, \r и т. Д.