编写一个python程序,将文件university.csv的内容逐行读取到列表中,根据用户输入的一个学校名,查询学校信息并以字典形式输出
时间: 2024-03-08 12:46:43 浏览: 166
python实现 列表转换为字典
5星 · 资源好评率100%
好的,以下是Python程序的代码:
```python
import csv
def read_csv_file(file_path):
university_list = []
with open(file_path, newline='', encoding='utf-8') as csvfile:
reader = csv.reader(csvfile)
for row in reader:
university_list.append(row)
return university_list
def search_university_info(university_name, university_list):
for university_info in university_list:
if university_name in university_info:
return {'name': university_info[0], 'location': university_info[1], 'type': university_info[2]}
return None
if __name__ == '__main__':
university_list = read_csv_file('university.csv')
university_name = input('请输入要查询的学校名称:')
university_info = search_university_info(university_name, university_list)
if university_info:
print(university_info)
else:
print('没有查询到该学校的信息。')
```
这个程序首先定义了两个函数,`read_csv_file`函数可以读取指定的CSV文件,将文件内容逐行读取到一个列表中,`search_university_info`函数则是根据用户输入的学校名称,在列表中查找对应的学校信息,并返回一个字典。
在`if __name__ == '__main__'`语句块中,程序首先调用`read_csv_file`函数读取CSV文件,并通过`input`函数获取用户输入的学校名称。然后,程序调用`search_university_info`函数查找学校信息,并将结果输出到控制台上。如果没有查询到对应的学校信息,程序会输出提示信息。
阅读全文