|

楼主 |
发表于 2024-7-23 09:00:31
|
显示全部楼层
import pandas as pd
# 读取生成数据.xlsx文件
df1 = pd.read_excel("原始数据.xlsx", usecols=["管线号", "焊缝类型", "工艺卡编号", "焊缝数量"])
# 只保留焊缝数量不为0的行
df1 = df1[df1["焊缝数量"] != 0]
# 读取提取数据.xlsx文件
df2 = pd.read_excel("原始数据.xlsx", usecols=["管线号", "焊缝类型2", "工艺卡编号2", "焊缝数量2"])
# 只保留焊缝数量2不为0的行
df2 = df2[df2["焊缝数量2"] != 0]
# 对df2中的列名进行重命名
df2.rename(columns={"焊缝类型2": "焊缝类型", "工艺卡编号2": "工艺卡编号", "焊缝数量2": "焊缝数量"}, inplace=True)
# 合并df1和df2
df_merge = pd.concat([df1, df2])
# 对相同的“管线号”,“焊缝类型”,“焊缝材质”进行分组,并对“焊缝数量”求和
df_merge = df_merge.groupby(["管线号", "焊缝类型", "工艺卡编号"], as_index=False).agg({"焊缝数量": "sum"})
# 根据“管线号”,“焊缝类型”,“焊缝材质”进行排序
df_merge.sort_values(by=["管线号", "焊缝类型", "工艺卡编号"], inplace=True)
# 输出到提取数据.xlsx文件
df_merge.to_excel("提取数据.xlsx", index=False)
实测ai这个可行,通过了 |
|