35 lines
1.0 KiB
Python
35 lines
1.0 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_csv_path, l2_csv_path, output_path):
|
||
"""
|
||
生成包含L1和L2两个sheet的Excel报表
|
||
:param l1_csv_path: L1等级统计结果CSV路径
|
||
:param l2_csv_path: L2等级统计结果CSV路径
|
||
:param output_path: 输出Excel文件路径
|
||
"""
|
||
# 读取CSV文件
|
||
df_l1 = pd.read_csv(l1_csv_path)
|
||
df_l2 = pd.read_csv(l2_csv_path)
|
||
|
||
# 创建Excel文件
|
||
with pd.ExcelWriter(output_path) as writer:
|
||
df_l1.to_excel(writer, sheet_name='L1等级组件', index=False)
|
||
df_l2.to_excel(writer, sheet_name='L2等级组件', index=False)
|
||
|
||
print(f"Excel报表生成成功：{output_path}")
|
||
|
||
if __name__ == "__main__":
|
||
if len(sys.argv) != 4:
|
||
print("用法：python generate_excel.py <l1_csv路径> <l2_csv路径> <输出Excel路径>")
|
||
sys.exit(1)
|
||
|
||
generate_excel(sys.argv[1], sys.argv[2], sys.argv[3])
|