0
点赞
收藏
分享

微信扫一扫

查询高校名

yongxinz 2022-02-01 阅读 70
python
def read_csv_to_lst(filename):
    """接收CSV文件名为参数,读取文件内容到二维列表,每行数据根据逗号切分为子列表,返回二维列表。"""
    with open(filename, 'r', encoding='utf-8') as f:
        university_lst = [line.strip().split(',') for line in f]
    return university_lst


def query_name(word, university_lst):
    """接收一个字符串和列表名为参数,从列表中查询学校名包含参数字符串的学校名,以列表形式返回。"""
    uni_name_lst = []
    for university in university_lst:
        if word in university[1]:
            uni_name_lst.append(university[1])
    return uni_name_lst


if __name__ == '__main__':
    file = 'university.csv'
    uni_lst = read_csv_to_lst(file)  # 获得高校信息二维列表
    key_word = input()  # 输入查询关键字
    ls = query_name(key_word, uni_lst)  # 查询包含关键的校名,得到列表
    print(*ls, sep='\n')  # 解包输出列表中的元素,用换行符做分隔 符,实现换行输出的效果
举报

相关推荐

0 条评论