代码概述 根据你的要求，我们将实现一个 Python 程序，利用**卷积计算**来处理 `RS` 数据（地表径流），并计算其总值 `R_surface`。该程序会读取 Excel 文件中的 `RS` 列，使用给定的单位线进行卷积运算，并输出结果。 --- ### ✅ 代码实现（卷积计算 `RS` 的总值 `R_surface`） ```python import pandas as pd import numpy as np import os # ==================== 1. 数据准备 ==================== # 读取Excel数据 data_path = r'C:\Users\80401\Desktop\BAOYU.xlsx' data = pd.read_excel(data_path, sheet_name='Sheet1') # 提取数据 RS = data['RS'].dropna().values # 地表径流数据（假设RS列为地表径流） unit_hydrograph = np.array([0, 40, 80, 130, 100, 80, 48, 20, 10, 5, 0]) # 10mm单位线 area = 553 # 流域面积 (km²) # 将地表径流RS从mm/3h转换为m³/s RS_m3s = RS * area * 1000 / (3 * 3600) # ==================== 2. 卷积计算 ==================== def calculate_convolution(RS, unit_hydrograph, area): """ 计算地表径流RS通过单位线的卷积，得到地表径流过程线。 参数: RS (array-like): 地表径流序列 (mm/3h) unit_hydrograph (array-like): 单位线 (m³/s) area (float): 流域面积 (km²) 返回: array-like: 卷积后的地表径流过程线 (m³/s) """ # 单位线缩放 (假设单位线是10mm净雨的单位线) unit_hydrograph_scaled = unit_hydrograph / 10 # 卷积计算 Q_RS = np.convolve(RS, unit_hydrograph_scaled, mode='full')[:len(RS)] return Q_RS # 执行卷积计算 Q_RS = calculate_convolution(RS_m3s, unit_hydrograph, area) # ==================== 3. 计算总值 R_surface ==================== R_surface = np.sum(Q_RS) # ==================== 4. 输出结果 ==================== print(f"卷积计算的地表径流总值 R_surface: {R_surface:.2f} m³") # ==================== 5. 保存结果到Excel ==================== # 创建输出目录 output_dir = os.path.expanduser('~/Desktop/case_results') os.makedirs(output_dir, exist_ok=True) # 保存结果到Excel output_path = os.path.join(output_dir, 'surface_runoff_result.xlsx') result_df = pd.DataFrame({ 'Time': data['T（i）'].dropna().values, # 假设时间列为'T（i）' 'RS (mm/3h)': RS, 'Q_RS (m³/s)': Q_RS }) result_df.to_excel(output_path, index=False) print(f"结果已保存到: {output_path}") ``` --- ### 💡 关键点解释 1. **数据提取**： - 假设 `RS` 列是地表径流数据； - 使用 `dropna()` 删除可能存在的缺失值，确保数据完整性。 2. **单位转换**： - 将 `RS` 从 `mm/3h` 转换为 `m³/s`，以便与单位线进行卷积运算。 3. **卷积计算**： - 使用 `np.convolve` 函数进行卷积运算，模拟地表径流过程线； - `mode='full'` 确保卷积结果长度与原始数据相同。 4. **总值计算**： - 使用 `np.sum()` 对卷积后的结果进行求和，得到 `R_surface`。 5. **结果保存**： - 将计算结果保存到一个新的 Excel 文件中，方便后续分析和查看。 --- ### 📌 注意事项 - **单位线的选择**：确保单位线与输入数据单位一致（本例假设为 10mm 净雨对应的单位线）。 - **时间列引用**：确保时间列的名称正确无误（如 `T（i）`），可以根据实际情况调整。 - **文件路径设置**：根据实际文件路径修改 `data_path` 和 `output_path`。 --- ### ✅ 知识点总结 1. **卷积操作**：用于模拟地表径流过程，是一种常用的水文模拟方法。 2. **单位转换**：确保不同物理量之间的单位一致性，避免计算误差。 3. **pandas 和 numpy 应用**：用于高效的数据处理和数值计算。 --- 如果文件结构或数据列有所不同，或有其他特殊需求，请提供更多信息以便进一步优化代码。