Python:在文本文件中搜索字符串并显示匹配的行

Question

我现在有这个程序，它允许用户从类别中进行选择(从文件中提取)。然后它将打印该文本文件中的大学或人员。

我接下来想要在代码中执行的操作是让用户从该文件中搜索特定字符串，它将显示具有该匹配字符串的大学和人员。它可以是该文件中的整个单词或字符串的一部分。

我需要有关搜索给定字符串或字符串的一部分并显示匹配类别(大学和人物)的帮助。

示例:

搜索:哦

输出:

大学:加州大学洛杉矶分校姓名:John Kelly

大学:孟菲斯大学姓名:Taylor Johnson

这是我当前的文本文件:

"UniversityName","ContactName"
"UCLA","John Kelly"
"UofFlorida","Mary Elizabeth"
"U of Memphis","Taylor Johnson"
"Harvard","Robert Fax"

这是我到目前为止对代码所做的事情:

def load_data(file_name):
    university_data=[]
    with open(file_name,'r') as rd_file:
        for line in rd_file.readlines():
            line=line.strip().split(',')
            T = line[0],line[1]
            university_data.append(T)
    return university_data

def main():
    filename='List.txt'
    university_data = load_data(filename)
    print('[1] University\n[2] Contact Name\n[3] Exit\n[4] Search')
    while True:

        choice=input('Enter choice 1/2/3? ')
        if choice=='1':
            for university in university_data:
                print(university[0])
        elif choice=='2':
            for university in university_data:
                print(university[1])
        elif choice =='3':
            print('Thank You')
            break
        elif choice =='4':
            print("Search Section Here")
        else:
            print('Invalid selection')

main()

Answer 1

根据您的数据:

university_data = [["UniversityName","ContactName"],
["UCLA","John Kelly"],
["UofFlorida","Mary Elizabeth"],
["U of Memphis","Taylor Johnson"],
["Harvard","Robert Fax"]]

使用str.join:

def search(key, data):
    for l in data:
        if key in ''.join(l):
            print('University: %s Name: %s' % (l[0], l[1]))
search('ohn', university_data)
# University: UCLA Name: John Kelly
# University: U of Memphis Name: Taylor Johnson