下面是一个编写的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
函数来将列表作为新列添加到数据框中。最后,我们打印了更新后的数据框。