Pandas -作为字典的列-如何始终提取第一个元素

import pandas as pd

# 创建包含字典列的DataFrame
data = {'col1': [{'a': 1, 'b': 2}, {'c': 3, 'd': 4}], 'col2': [5, 6]}
df = pd.DataFrame(data)

# 定义一个函数，用于提取字典列的第一个元素
def extract_first_element(dic):
    return next(iter(dic.values()))

# 使用apply函数调用提取函数，并将结果赋值给新的列
df['first_element'] = df['col1'].apply(lambda x: extract_first_element(x))

# 打印DataFrame
print(df)

           col1  col2 first_element
0  {'a': 1, 'b': 2}     5             1
1  {'c': 3, 'd': 4}     6             3