41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
import pandas as pd
|
||
import numpy as np
|
||
import sys
|
||
|
||
# 兼容numpy版本
|
||
try:
|
||
np._get_promotion_state = lambda *args, **kwargs: 0
|
||
except:
|
||
pass
|
||
|
||
def generate_excel(l1_oops_csv, l2_oops_csv, l1_exit_csv, l2_exit_csv, output_path):
|
||
"""
|
||
生成包含4个sheet的Excel报表
|
||
:param l1_oops_csv: L1等级Oops率统计结果CSV路径
|
||
:param l2_oops_csv: L2等级Oops率统计结果CSV路径
|
||
:param l1_exit_csv: L1等级退出率统计结果CSV路径
|
||
:param l2_exit_csv: L2等级退出率统计结果CSV路径
|
||
:param output_path: 输出Excel文件路径
|
||
"""
|
||
# 读取CSV文件
|
||
df_l1_oops = pd.read_csv(l1_oops_csv)
|
||
df_l2_oops = pd.read_csv(l2_oops_csv)
|
||
df_l1_exit = pd.read_csv(l1_exit_csv)
|
||
df_l2_exit = pd.read_csv(l2_exit_csv)
|
||
|
||
# 创建Excel文件
|
||
with pd.ExcelWriter(output_path) as writer:
|
||
df_l1_oops.to_excel(writer, sheet_name='L1组件Oops率', index=False)
|
||
df_l2_oops.to_excel(writer, sheet_name='L2组件Oops率', index=False)
|
||
df_l1_exit.to_excel(writer, sheet_name='L1组件退出率', index=False)
|
||
df_l2_exit.to_excel(writer, sheet_name='L2组件退出率', index=False)
|
||
|
||
print(f"Excel报表生成成功：{output_path}")
|
||
|
||
if __name__ == "__main__":
|
||
if len(sys.argv) != 6:
|
||
print("用法：python generate_excel.py <l1_oops_csv路径> <l2_oops_csv路径> <l1_exit_csv路径> <l2_exit_csv路径> <输出Excel路径>")
|
||
sys.exit(1)
|
||
|
||
generate_excel(sys.argv[1], sys.argv[2], sys.argv[3], sys.argv[4], sys.argv[5])
|