以前写的一个简单拼音分词程序。"tianqibucuo" -> "tian qi bu cuo"

pinyin_seg.py

#-*-coding:utf-8-*-

def seg(pinyin):
    sm = ['b', 'c', 'd', 'f', 'g', 'h', 'j', 'k', 'l', 'm', 'n', 'p', 'q', 'r', 's', 't', 'w', 'x', 'y', 'z', 'sh', 'zh', 'ch']
    ymd = {'a': {'i':None, 'o':None, 'n': {'g': None}}, 
           'e': {'i':None, 'r':None, 'n': {'g': None}}, 
           'i': {'e':None, 'a':{'n':{'g':None}, 'o':None}, 'u':None, 'o':{'n':{'g':None}}, 'n':{'g':None}},
           'o': {'u':None, 'n':{'g':None}},
           'u': {'a':{'i':None, 'n':{'g':None}}, 'e':None, 'i':None, 'o':None}
          }
    result, i = '', 0
    while i < len(pinyin)-1:
        if pinyin[i] in ['s', 'z', 'c'] and pinyin[i+1] == 'h':
            result += ' ' + pinyin[i] + 'h'
            i += 2
        elif pinyin[i] in sm:
            result += ' ' + pinyin[i]
            i += 1
        else:
            result += ' '
        tmp, tmpd = '', ymd
        while i < len(pinyin):
            tmpd = tmpd[pinyin[i]]
            tmp += pinyin[i]
            i += 1            
            if not (i < len(pinyin) and tmpd and tmpd.has_key(pinyin[i])):
                break
        result += tmp
    return result.strip()

if __name__ == "__main__":
    print seg("congqianxiedeyigejiandandepinyinfencichengxu")