Question

PYTHON PROGRAMMING:

I have this program right now where it allows users to choose from a category(pulling from the file). Then it will print the University or people from that text file.

What I want to do next on my code is for users to search for a specific string from that file and it will display both the University and People that have that matching string. It can be the whole word or part of the string from that file.

Example:

search: ohn

output:

University: UCLA Name: John Kelly

University: U of Memphis Name: Taylor Johnson

Here is my current text file:

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

This is what I've done so far with my code:

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

ANSWER CODE:

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 find_name(university_data,sub_str):
for university in university_data:
res=university[1].find(sub_str)
if(res>0):
print('University:',university[0],'Name:',university[1])
  
  

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/4? ')
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':
   sub_str=input("Enter any substring to search in name:")
   find_name(university_data,sub_str);
  
else:
print('Invalid selection')

main()

OUTPUT: