使用spaCy从文本中提取与给定日期相关的日期

import spacy
from spacy.matcher import Matcher
from spacy.tokens import Span
import re

nlp = spacy.load("en_core_web_sm")

def extract_dates(text, target_date):
    # 将文本传递给spaCy进行处理
    doc = nlp(text)
    
    # 定义一个模式匹配器
    matcher = Matcher(nlp.vocab)
    
    # 定义日期模式
    pattern = [{"LOWER": {"IN": ["on", "in", "at"]}}, {"ENT_TYPE": "DATE"}]
    
    # 将模式添加到匹配器中
    matcher.add("DATE_PATTERN", None, pattern)
    
    # 在文档中找到匹配的日期
    matches = matcher(doc)
    
    # 提取匹配的日期
    extracted_dates = []
    for match_id, start, end in matches:
        span = Span(doc, start, end)
        extracted_dates.append(span.text)
    
    # 过滤出与给定日期相关的日期
    filtered_dates = []
    for date in extracted_dates:
        if re.search(r"\b{}\b".format(target_date), date, re.IGNORECASE):
            filtered_dates.append(date)
    
    return filtered_dates

text = "I have a meeting on January 1st, 2022 and another meeting on December 31st, 2021."
target_date = "January 1st, 2022"
extracted_dates = extract_dates(text, target_date)
print(extracted_dates)