python 将带数字的拼音转换为带声调的拼音

最终实现结果如图

最后附上代码，

from pypinyin import pinyin, Style

# 传入汉字和带数字的拼音,然后返回的值是带音调的拼音
def text_tone2pinyin(txt, tone):

    tone_list = pinyin(txt, heteronym=True, style=Style.TONE3)[0]
    pin_list = pinyin(txt, heteronym=True)[0]

    for i in range(len(tone_list)):
        if tone == tone_list[i]:
            return pin_list[i]

 最后直接调用函数即可