为Pandas dataframe单元格中的列表元素编写if语句

import pandas as pd

# 创建一个包含列表的dataframe
df = pd.DataFrame({'col1': [[1, 2, 3], [4, 5, 6], [7, 8, 9]]})

# 定义一个函数，用于判断列表元素是否满足条件
def check_list(lst):
    if 2 in lst:
        return '包含2'
    else:
        return '不包含2'

# 使用apply函数和lambda表达式应用判断函数到每个单元格
df['col1'] = df['col1'].apply(lambda x: check_list(x))

print(df)

   col1
0  包含2
1  不包含2
2  不包含2