使用函數從文本文件中檢索特定數據。 Python 3

Question

有一個外部文本文件，它具有列中的信息。

例如：

文本文件看起來是這樣的：

123 1 645 Kallum Chris Gardner 
143 2 763 Josh Brown Sinclar 

等

現在的數字 「1」 和 「2」 是十年。我需要編寫一個程序，獲取當年的輸入並打印出關於個人的其他信息。

所以我會在程序中輸入「2」，'143 2 763 Josh Brown Sinclar'會被打印出來。

到目前爲止，我得到了這樣的代碼。我如何繼續前進？

def order_name(regno, year, course, first_name, middle_name, last_name=None): 
    if not last_name: 

      last_name = middle_name 
    else: 

      first_name = first_name, middle_name 
    return (last_name, first_name,regno, course, year) 

Answer 1

你可以做這樣的事情：

f = open('your_file.txt') 
lines = f.readlines() 
res = [x for x in lines if str(year) in x.split()[1]] 
print res