Question

我现在有这个程序，它允许用户从类别中选择（从文件中拉出）。然后它将打印该文本文件中的大学或人员。

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

在搜索给定的字符串或字符串的一部分并显示匹配的类别（大学和人）时，我需要帮助。

示例：

搜索：哦

输出：

大学：UCLA名称：J ohn Kelly

大学：孟菲斯大学名称：Taylor J ohn son

这是我当前的文本文件：

"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

Answer 2

与@Chris的上一个答案类似，如果要忽略搜索词的大写或小写，可以在lower()函数中添加search。

def search(key, mydata):
    for l in mydata:
        if key in (''.join(l)).lower():
            print('University: %s Name: %s' % (l[0], l[1]))

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

2 个答案: