如何从pandas Dataframe中包含数组的单元格中提取一个值？

import pandas as pd

# 创建包含数组的DataFrame
data = {'col1': [[1, 2, 3], [4, 5, 6], [7, 8, 9]],
        'col2': [[10, 11, 12], [13, 14, 15], [16, 17, 18]]}
df = pd.DataFrame(data)

# 定义提取值的函数
def extract_value(cell):
    return cell[0]  # 提取数组中的第一个元素

# 使用apply和lambda函数提取值
df['extracted_value'] = df['col1'].apply(lambda x: extract_value(x))

# 打印结果
print(df)

        col1          col2  extracted_value
0  [1, 2, 3]  [10, 11, 12]                1
1  [4, 5, 6]  [13, 14, 15]                4
2  [7, 8, 9]  [16, 17, 18]                7