如何在Python Pandas中将FOR循环中的字符串更改为列表中的元素

import pandas as pd

# 创建包含字符串的DataFrame列
df = pd.DataFrame({'col1': ['str1', 'str2', 'str3']})

# 定义函数，将字符串更改为列表中的元素
def change_string_to_list(string):
    # 假设列表为['element1', 'element2', 'element3']
    lst = ['element1', 'element2', 'element3']
    # 返回列表中对应的元素
    return lst[int(string[-1]) - 1]

# 使用apply函数将函数应用于DataFrame列
df['col1'] = df['col1'].apply(change_string_to_list)

# 打印结果
print(df)

       col1
0  element1
1  element2
2  element3