16.插入列(用文心一言写Python代码,智能分析Excel数据)

李光朱课程 2024-03-28 00:23:47
案例导入

如下图,在“电话补贴”那列前面加一列“交通补贴”,该列的均值为“600”。然后工资总额那列的值加600元。

图1.原始数据表

文心一言提示词

打开百度“文心一言”,输入提示词:“你是pandas专家,文件路径为:D:/Samples/各部门人员工资.xlsx。该Excel文件的第一个工作表中A-F列为给定数据。用pandas导入Excel文件的数据,导入前12行数据,引擎为"openpyxl"。在导入的数据中,在第4列前面插入一列“交通补贴”,该列的值均为600。然后给工资总额加600元。输出修改后的表数据。添加代码注释。”

显示结果如下:

首先,确保你已经安装了pandas和openpyxl库。如果没有,请使用pip安装它们:

pip install pandas openpyxl

以下是导入Excel文件数据、添加“交通补贴”列、修改工资总额,并输出修改后表数据的Python代码示例:

import pandas as pd# 设置文件路径file_path = 'D:/Samples/各部门人员工资.xlsx'# 使用pandas的read_excel函数读取Excel文件# 参数'sheet_name=0'表示读取第一个工作表# 参数'nrows=12'表示只读取前12行数据# 参数'engine='openpyxl''指定使用openpyxl引擎读取Excel文件df = pd.read_excel(file_path, sheet_name=0, nrows=12, engine='openpyxl')# 在第4列(索引为3,因为索引从0开始)前面插入一列“交通补贴”# 该列的值均设为600,dtype设置为intdf.insert(3, '交通补贴', 600, allow_duplicates=False)# 给工资总额(假设是最后一列,列名为'工资总额')加600元# 这里假设'工资总额'是数据框的最后一列df['工资总额'] += 600# 输出修改后的表数据print(df)# 如果需要,可以将修改后的数据保存回Excel文件df.to_excel('D:/Samples/修改后的各部门人员工资.xlsx', index=False, engine='openpyxl')

在上面的代码中,file_path变量存储了Excel文件的路径。pd.read_excel函数用于读取Excel文件中的数据,其中sheet_name=0表示读取第一个工作表,nrows=12指定只读取前12行数据,engine='openpyxl'指定使用openpyxl作为读取引擎。

使用insert函数在第4列(索引为3)前面插入新列“交通补贴”,并将该列所有值设置为600。allow_duplicates参数设置为False,以确保列名不会重复。

最后,假设“工资总额”是数据框的最后一列,我们直接给这一列的所有值加600元。然后,使用print函数输出修改后的数据框df。

请确保“工资总额”列名与你的Excel文件中的实际列名一致,如果不一致,请替换为正确的列名。如果工资总额不是最后一列,你需要指定正确的列索引或列名来修改该列的值。

Python编辑器运行

图2.Python运行结果

图3.生成的新文件数据

本合集使用的工具如下:

文心一言:3.5版本

Python编辑器:3.9.0

申明:以上结果皆为实际运行结果,如遇到无法运行可在评论区留言。部分数据来自网络,如有侵权请私信告知,感谢!

2 阅读:21

李光朱课程

简介:感谢大家的关注