在Pandas列中搜索list的元素，如果匹配，则将这些元素返回到新列

import pandas as pd

# 创建一个包含list的列
df = pd.DataFrame({'column_name': [['element1', 'element2'], ['element3', 'element4'], ['element5', 'element6']]})

# 定义一个函数，用于搜索并返回匹配的元素
def search_list(row):
    search_list = ['element1', 'element3']  # 要搜索的元素列表
    matched_elements = [element for element in row if element in search_list]  # 使用列表推导式搜索匹配的元素
    return matched_elements

# 使用apply函数和lambda表达式将匹配的元素添加到新列中
df['matched_elements'] = df['column_name'].apply(lambda row: search_list(row))

print(df)

         column_name matched_elements
0  [element1, element2]       [element1]
1  [element3, element4]       [element3]
2  [element5, element6]               []