下面是一个编写的Python函数,用于接收一个列表和一个数据框的一列,并根据该列表添加一个新的列。
import pandas as pd
def add_column_from_list(df, column_name, lst):
    df[column_name] = lst
    return df
# 示例用法
# 创建一个示例数据框
data = {'Name': ['Tom', 'Nick', 'John', 'Sam'],
        'Age': [20, 25, 30, 35]}
df = pd.DataFrame(data)
# 创建一个示例列表
new_column = ['A', 'B', 'C', 'D']
# 调用函数添加新列
df = add_column_from_list(df, 'NewColumn', new_column)
# 打印结果
print(df)
运行上述代码将输出以下结果:
  Name  Age NewColumn
0  Tom   20         A
1 Nick   25         B
2 John   30         C
3  Sam   35         D
在这个示例中,我们定义了一个add_column_from_list函数,它接收一个数据框df,一个列名column_name和一个列表lst。然后,我们使用df[column_name] = lst将列表lst作为新列添加到数据框df中。最后,函数返回更新后的数据框。
在示例用法中,我们创建了一个示例数据框和一个示例列表,并调用了add_column_from_list函数来将列表作为新列添加到数据框中。最后,我们打印了更新后的数据框。