如何将SQL Partition By逻辑迁移到代码中计算DataTable的Weight列值

AI 小北

2026-9-29

SQL权重计算逻辑迁移C#实现方案评估

现有实现思路不合理，核心问题如下：

冗余计算开销过高：你在逐行遍历的循环内部，每次都对全表做分组、复制生成多份子DataTable，时间复杂度达到O(n²)，数据量稍大就会出现严重的性能卡顿。
核心逻辑缺失：仅完成了按MandateCode拆分数据的操作，完全没有实现SQL中「先按分组求和、再逐行计算权重」的核心逻辑，也没有给每行的Weight字段赋值。
内存浪费：分组后调用CopyToDataTable生成了多份冗余的子表数据，额外占用内存，完全不需要拆分原DataTable。

对应SQL逻辑的正确实现方案

你原来的SQL是按MandateCode作为分区维度计算每个分组的BaseVal总和，再逐行计算当前行占分组总和的比例，总和为0时权重取0，对应C#实现如下：

// 1. 提前给DataTable新增Weight列，类型和业务需要的精度匹配即可，这里用decimal避免精度丢失
if (!datatable.Columns.Contains("Weight"))
{
    datatable.Columns.Add("Weight", typeof(decimal));
}

// 2. 先全局计算每个MandateCode对应的BaseVal总和，只计算1次，缓存到字典
var mandateSumDict = datatable.AsEnumerable()
    .GroupBy(row => row.Field<int>("MandateCode"))
    .ToDictionary(
        groupKey => groupKey.Key,
        group => group.Sum(row => row.Field<decimal>("BaseVal"))
    );

// 3. 逐行计算权重赋值
foreach (var row in datatable.AsEnumerable())
{
    int currentMandateCode = row.Field<int>("MandateCode");
    decimal groupSum = mandateSumDict[currentMandateCode];
    row["Weight"] = groupSum == 0 ? 0 : row.Field<decimal>("BaseVal") / groupSum;
}

该方案时间复杂度为O(n)，无冗余计算和内存开销，完全对齐原SQL的计算逻辑。