spaCy nlp - заменить строку - PullRequest
Новая
       50

spaCy nlp - заменить строку

0 голосов
/ 11 апреля 2020 
import spacy
import en_core_web_sm
import re


nlp = en_core_web_sm.load()
document_string= 'Electronically signed by : John Douglas.; Jun 13 2018 11:13AM CST, Adam Smith.'
nlp_doc = nlp(document_string)
from spacy.matcher import Matcher
matcher = Matcher(nlp.vocab)
pattern = [{'POS': 'PROPN'}, {'POS': 'PROPN'}]
matcher.add('FULL_NAME', None, pattern)
matches = matcher(nlp_doc)
for match_id, start, end in matches:
     span = nlp_doc[start:end]
     names = span.text
     print(span.text)

Вывод: 

John Douglas
Adam Smith

Мне нужно заменить его на [скрытый] и напечатать document_string с показом скрытых значений (предыдущий вывод)

Требуемый выход: 

Electronically signed by : [hidden].; Jun 13 2018 11:13AM CST, [hidden].

1 Ответ

0 голосов
/ 11 апреля 2020

Вы можете просто использовать .replace() здесь. 

new_doc = nlp_doc.text
names = []

pattern = [{'POS': 'PROPN'}, {'POS': 'PROPN'}] 
matcher.add('FULL_NAME', None, pattern) 
matches = matcher(nlp_doc) 

for match_id, start, end in matches: 
    span = nlp_doc[start:end] 
    names.append(span.text) 

for name in names: 
    new_doc = new_doc.replace(name,'[hidden]')

Результат: 

In [114]: new_doc
Out[114]: 'Electronically signed by : [hidden].; Jun 13 2018 11:13AM CST, [hidden].'
...