根据名字拼音首字母简拼查找名字

#python3.6
from xpinyin import Pinyin
#导入xpinyin库，可以把汉字转换成拼音

name_dict = dict()
#设置一个保存名字和名字拼音首字母的字典，例：d={"ww":"王五"}
fo=open("/home/gty/文档/data/namelist.txt", "r")
# 读取文档，请读者自行更换路径

def namespy(names):
    p = Pinyin()
    s = p.get_initials(names, u'').lower()
    #获取名字的简拼，"王五"->"ww"
    name_dict.setdefault(s, []).append(names)
    #更新字典,这里字典的值为列表
    return s
#将名字加入字典，拼音首字母为键

for line in fo:
    namespy(line.strip())
    # print(line.strip(),end="")
#遍历文件的每一行

str = input("请输入名字首字母的组合：")
if str in name_dict:
    print("查找到以下名字")
    for i in name_dict[str]:
        print(i,"\t",end="")
else:
    print("没找到")