89 lines
3.0 KiB
Python
89 lines
3.0 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
列计算模板
|
||
对列进行加减乘除等计算操作
|
||
"""
|
||
|
||
import pandas as pd
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
def main():
|
||
# 参数配置
|
||
file_path = "{file}"
|
||
output_path = "{output}"
|
||
operation = "{operation}" # 操作类型: add/subtract/multiply/divide/custom
|
||
|
||
# 操作参数
|
||
column1 = "{column1}" # 第一列
|
||
column2 = "{column2}" # 第二列（可选）
|
||
result_column = "{result_column}" # 结果列名
|
||
value = {value} # 常数值（可选）
|
||
formula = "{formula}" # 自定义公式（可选）
|
||
|
||
# 读取文件
|
||
df = pd.read_excel(file_path)
|
||
|
||
print(f"原始数据: {{len(df)}} 行")
|
||
|
||
# 执行计算
|
||
if operation == "add":
|
||
if column2:
|
||
df[result_column] = df[column1] + df[column2]
|
||
print(f"计算: {{column1}} + {{column2}} = {{result_column}}")
|
||
else:
|
||
df[result_column] = df[column1] + value
|
||
print(f"计算: {{column1}} + {{value}} = {{result_column}}")
|
||
|
||
elif operation == "subtract":
|
||
if column2:
|
||
df[result_column] = df[column1] - df[column2]
|
||
print(f"计算: {{column1}} - {{column2}} = {{result_column}}")
|
||
else:
|
||
df[result_column] = df[column1] - value
|
||
print(f"计算: {{column1}} - {{value}} = {{result_column}}")
|
||
|
||
elif operation == "multiply":
|
||
if column2:
|
||
df[result_column] = df[column1] * df[column2]
|
||
print(f"计算: {{column1}} × {{column2}} = {{result_column}}")
|
||
else:
|
||
df[result_column] = df[column1] * value
|
||
print(f"计算: {{column1}} × {{value}} = {{result_column}}")
|
||
|
||
elif operation == "divide":
|
||
if column2:
|
||
df[result_column] = df[column1] / df[column2]
|
||
print(f"计算: {{column1}} ÷ {{column2}} = {{result_column}}")
|
||
else:
|
||
df[result_column] = df[column1] / value
|
||
print(f"计算: {{column1}} ÷ {{value}} = {{result_column}}")
|
||
|
||
elif operation == "custom" and formula:
|
||
# 自定义公式（简单实现）
|
||
try:
|
||
# 替换列名为实际的 Series
|
||
context = {{col: df[col] for col in df.columns}}
|
||
df[result_column] = eval(formula, {{'pd': pd}}, context)
|
||
print(f"自定义公式: {{formula}} = {{result_column}}")
|
||
except Exception as e:
|
||
print(f"自定义公式执行失败: {{e}}")
|
||
sys.exit(1)
|
||
|
||
# 保存结果
|
||
df.to_excel(output_path, index=False)
|
||
|
||
print(f"\n计算完成")
|
||
print(f"输出文件: {{output_path}}")
|
||
|
||
# 显示统计信息
|
||
if result_column in df.columns:
|
||
print(f"\n结果列统计 ({{result_column}}):")
|
||
print(f" 最小值: {{df[result_column].min():.2f}}")
|
||
print(f" 最大值: {{df[result_column].max():.2f}}")
|
||
print(f" 平均值: {{df[result_column].mean():.2f}}")
|
||
print(f" 总和: {{df[result_column].sum():.2f}}")
|
||
|
||
if __name__ == "__main__":
|
||
main()
|