有沒有糾正大小寫的python拼寫糾正庫？

Question

我有拼寫正確但全部小寫的字符串（第一個字符除外），我想更正大寫（英語 - 所以基本上只是事物的名稱......）。 我嘗試了 pyspellcheck、autocorrect 和 symspellpy，它們不考慮大寫 afaik。

因此，例如字符串'And then we went to see frank from england to have a beer with him.' 應該更正為'And then we went to see Frank from England to have a beer with him.' .

你知道任何可以做到這一點的圖書館嗎？

Answer 1

你可以用 spaCy 做到這一點：

import spacy
nlp=spacy.load('en_core_web_md')


def capitalize_ent(text):
    title_text=text.title()
    print(text)
    doc=nlp(title_text)
    words=[]
    for x in doc:
        if nlp(x.text).ents:
            words.append(x.text)
    for word in words:
        text=text.replace(word.lower(),word)
    return text
            

不要忘記下載spaCy語言model：

python -m spacy download en_core_web_md