对于Pandas Dataframe中的每个元素，将列表元素拆分为单词的子列表

import pandas as pd
import re

def split_words(element):
    if isinstance(element, list):
        words = []
        for item in element:
            words.extend(re.findall(r'\w+', str(item)))
        return words
    else:
        return re.findall(r'\w+', str(element))

def split_words_in_dataframe(df):
    for column in df.columns:
        df[column] = df[column].apply(split_words)
    return df

# 示例数据
data = {'A': [['apple', 'banana'], ['cat', 'dog']],
        'B': [['elephant', 'fox'], ['grape', 'horse']]}
df = pd.DataFrame(data)

# 拆分列表元素为单词的子列表
df = split_words_in_dataframe(df)

print(df)

          A           B
0  [apple, banana]  [elephant, fox]
1     [cat, dog]  [grape, horse]