在两个Pandas Dataframe列之间查找匹配关键字之前和之后的单词，并在新列中返回

import pandas as pd
import re

def find_words_before_and_after(df, column1, column2, keyword):
    # 创建新列
    df['new_column'] = ""

    # 遍历DataFrame的行
    for index, row in df.iterrows():
        # 获取两个列的值
        text1 = row[column1]
        text2 = row[column2]
        
        # 使用正则表达式查找匹配关键字之前和之后的单词
        pattern = r"\b(\w*{}(?:\w*)?)\b".format(re.escape(keyword))
        matches1 = re.findall(pattern, text1)
        matches2 = re.findall(pattern, text2)
        
        # 拼接找到的单词
        words = " ".join(matches1 + matches2)
        
        # 将单词存储在新列中
        df.at[index, 'new_column'] = words

    return df

# 创建示例DataFrame
df = pd.DataFrame({'text1': ['Hello world', 'This is a test'], 'text2': ['World, hello', 'Test, test']})

# 调用函数查找匹配关键字之前和之后的单词
df = find_words_before_and_after(df, 'text1', 'text2', 'test')

# 打印结果
print(df)