参考资料：
    【python 走进NLP】搜索提示功能前缀字典树
    巧用 Trie 树实现搜索引擎关键词提示功能
    python利用Trie(前缀树)实现搜索引擎中关键字输入提示（学习Hash Trie和Double-array Trie）
    【python 走进NLP】搜索提示功能前缀字典树 - 图1
    站内搜索是一个网站的基本功能，一个好的搜索提示也能很好的提升用户体验，提高用户找到自己需要的东西的效率。下面基于字典树实现一个简单的中文前缀搜索提示功能。
    用户输入的时候自动提示。
    image.png

    2. # -*- encoding=utf-8 -*-
    5. # 导入包
    6. from pytrie import StringTrie
    8. # 自定义字典树类
    9. class Suggester(object):
    10.     def __init__(self):
    11.         self.trie = None
    12.         self.trie = StringTrie()
    13.     def update_trie(self, word_list):
    14.         for word in word_list:
    15.             word = word.lower()
    16.             self.trie[word] = word
    18.     def search_prefix(self, prefix):
    19.         return self.trie.values(prefix=prefix)
    22. # 建立前缀字典树
    23. def build_prefix_tree(wordlist):
    24.     word_list = open(wordlist).read().splitlines()
    25.     suggester = Suggester()
    26.     suggester.update_trie(word_list)
    27.     return suggester
    33. if __name__ == '__main__':
    35.     sug = Suggester()
    36.     sug.update_trie(['意味深长','意想不到','意气用事','意气风发','意兴阑珊','意气高昂'])
    38.     print(sug.search_prefix('意'))
    39.     print(sug.search_prefix('意气'))

    运行结果：

    1. E:\laidefa\python.exe E:/短信报警/SearchSuggestion-master/backend/实现搜索提示功能.py
    2. ['意想不到', '意兴阑珊', '意味深长', '意气风发', '意气高昂', '意气用事']
    3. ['意气风发', '意气高昂', '意气用事']
    5. Process finished with exit code 0

    更复杂的搜索提示功能：支持拼音，首字母拼音，中文等。类似优酷，酷狗等搜索提示功能。完美实现。
    image.png
    image.png
    安装包：

    1. pip install pytrie
    2. pip install pandas
    3. pip install pypinyin

    image.png
    搜索提示功能实现：

    1. # -*- encoding=utf-8 -*-
    3. # 导入包
    4. from pytrie import StringTrie
    5. import pypinyin
    6. import pandas as pd
    7. import time
    10. # 文本转拼音
    11. def pinyin(text):
    12.     """
    13.     :param text: 文本
    14.     :return: 文本转拼音
    15.     """
    16.     gap = ' '
    17.     piny = gap.join(pypinyin.lazy_pinyin(text))
    18.     return piny
    21. # 获取拼音的每个首字母
    22. def get_every_word_first(text):
    23.     """
    24.     :param text:文本
    25.     :return: 返回拼音首字母
    26.     """
    27.     return ''.join([i[0] for i in pinyin(text).split(' ')])
    31. # 获取拼音的第一个首字母
    32. def get_all_pinying(text):
    33.     """
    34.         :param text: 文本
    35.         :return: 文本转拼音
    36.         """
    37.     gap = ''
    38.     piny = gap.join(pypinyin.lazy_pinyin(text))
    40.     return piny
    44. # 自定义字典树类
    45. class Suggester(object):
    46.     def __init__(self):
    47.         self.trie = None
    48.         self.trie = StringTrie()
    50.     def update_trie(self, word_list):
    51.         for word in word_list:
    52.             word = word.lower()
    53.             # 拼音提取
    54.             word_pinyin1=get_every_word_first(word)
    55.             word_pinyin2=get_all_pinying(word)
    57.             # 拼音建立字典树
    58.             self.trie[word] = word
    59.             self.trie[word_pinyin1]=word_pinyin1
    60.             self.trie[word_pinyin2] = word_pinyin2
    63.     def search_prefix(self, prefix):
    64.         return self.trie.values(prefix=prefix)
    68. # 构建字典树
    69. def build_all_trie(wordlist):
    70.     """
    71.     :param wordlist: 关键词列表
    72.     :return: 字典树和映射数据集
    73.     """
    74.     sug = Suggester()
    75.     sug.update_trie(wordlist)
    76.     data = pd.DataFrame({"word": wordlist})
    77.     data['pinyin1'] = data['word'].apply(lambda x: get_every_word_first(x))
    78.     data['pinyin2'] = data['word'].apply(lambda x: get_all_pinying(x))
    80.     return sug,data
    83. # 判断字符串只包含中文
    84. def check_contain_chinese(check_str):
    85.     flag = True
    86.     for ch in check_str:
    87.         if u'\u4e00' >= ch or ch >= u'\u9fff':
    88.             flag =  False
    89.     return flag
    94. # 关键词搜索提示查询
    95. def get_tips_word(sug,data,s):
    96.     """
    97.     :param sug: 字典树
    98.     :param data: 中文和英文映射数据集
    99.     :param s: 搜索词
    100.     :return: 返回搜索提示词
    101.     """
    102.     try:
    103.         if len(s)>0:
    104.             # 判断输入是否只包含中文，若只中文，按中文查
    105.             if check_contain_chinese(s) is True:
    106.                 # 输出结果
    107.                 kk = sug.search_prefix(s)
    108.                 result3 = data[data['word'].isin(kk)]
    109.                 result6 = list(set(result3['word']))
    110.                 return result6
    112.             # 若不是只包含中文，转换为英文去查询
    113.             else:
    114.                 s1=get_all_pinying(s)
    115.                 kk = sug.search_prefix(s1)
    116.                 result1 = data[data['pinyin1'].isin(kk)]
    117.                 result2 = data[data['pinyin2'].isin(kk)]
    118.                 result3 = data[data['word'].isin(kk)]
    119.                 result4 = result1.append(result2, ignore_index=True)
    120.                 result5 = result3.append(result4, ignore_index=True)
    121.                 # 输出结果
    122.                 result6 = list(set(result5['word']))
    123.                 return result6
    124.         else:
    125.             return
    127.     except Exception as e:
    128.         print("{0}".format(str(e)))
    133. if __name__ == '__main__':
    135.     wordlist=['意味深长','意想不到','意气用事','意气风发','意兴阑珊','意气高昂','意气相投','巴黎恋人','巴黎圣母院','巴黎宝贝']
    137.     # 构造字典树
    138.     sug, data = build_all_trie(wordlist)
    139.     time1=time.time()
    142.     # 搜索词
    143.     s='b'
    145.     result=get_tips_word(sug,data,s)
    146.     print(result)
    148.     time2=time.time()
    149.     print('总共耗时：' + str(time2- time1) + 's')

    运行结果：

    1. E:\laidefa\python.exe E:/短信报警/SearchSuggestion-master/backend/实现搜索提示功能.py
    2. ['巴黎圣母院', '巴黎宝贝', '巴黎恋人']
    3. 总共耗时：0.002985239028930664s
    5. Process finished with exit code 0