浅谈利用LINQ进行分组统计

让我们来生成要统计的数据，如下所示：

让客户满意是我们工作的目标，不断超越客户的期望值来自于我们对这个行业的热爱。我们立志把好的技术通过有效、简单的方式提供给客户，将通过不懈努力成为客户在信息化领域值得信任、有价值的长期合作伙伴，公司提供的服务项目有：空间域名、网站空间、营销软件、网站建设、哈密网站维护、网站推广。

 
 
 

  
  
  
IEnumerable, double>> GetTuples(int n)  
  
  
  
{  
  
  
  
  var tuples = new Tuple, double>[n];  
  
  
  
  var rand = new Random();  
  
  
  
  for (int k = 1, i = 0; i < n; i++)  
  
  
  
  {  
  
  
  
    var r = rand.Next(n);  
  
  
  
    k += (r >= n - 3) ? 2 : ((r >= n - 9) ? 1 : 0);  
  
  
  
    tuples[i] = new Tuple, double>(k, rand.NextDouble());  
  
  
  
  }  
  
  
  
  return tuples;  
  
  
  
} 
 
 
 

该方法生成 n 项已经排好序的数据。

现在，让我们来按关键字分组，并统计每组的个数和平均值。

首先，使用 C# 的 foreach 循环，如下所示：

 
 
 

  
  
  
IEnumerable, int, double>> ForEach(IEnumerable, double>> tuples)  
  
  
  
{  
  
  
  
  var result = new List, int, double>>();  
  
  
  
  var count = 0;  
  
  
  
  var sum = 0.0;  
  
  
  
  int? key = null;  
  
  
  
  foreach (var v in tuples)  
  
  
  
  {  
  
  
  
    if (key != v.Item1)  
  
  
  
    {  
  
  
  
      if (key != null) result.Add(new Tuple, int, double>(key.Value, count, sum / count));  
  
  
  
      sum = count = 0;  
  
  
  
      key = v.Item1;  
  
  
  
    }  
  
  
  
    count++;  
  
  
  
    sum += v.Item2;  
  
  
  
  }  
  
  
  
  if (key != null) result.Add(new Tuple, int, double>(key.Value, count, sum / count));  
  
  
  
  return result;  
  
  
  
} 
 
 
 

这种方法有个***的缺点就是在 foreach 循环结束之后还要进行一次统计，闻到了代码的“坏味道”。

那么，就让我们来重构吧，这次，使用迭代器进行循环：

 
 
 

  
  
  
IEnumerable, int, double>> Iterate(IEnumerable, double>> tuples)  
  
  
  
{  
  
  
  
  var result = new List, int, double>>();  
  
  
  
  var count = 0;  
  
  
  
  var sum = 0.0;  
  
  
  
  int? key = null;  
  
  
  
  for (var iter = tuples.GetEnumerator(); ; count++, sum += iter.Current.Item2)  
  
  
  
  {  
  
  
  
    var hasValue = iter.MoveNext();  
  
  
  
    if (!hasValue || key != iter.Current.Item1)  
  
  
  
    {  
  
  
  
      if (key != null) result.Add(new Tuple, int, double>(key.Value, count, sum / count));  
  
  
  
      if (!hasValue) break;  
  
  
  
      sum = count = 0;  
  
  
  
      key = iter.Current.Item1;  
  
  
  
    }  
  
  
  
  }  
  
  
  
  return result;  
  
  
  
} 
 
 
 

这样，就消灭了“坏味道”。

注意，以上两种方法都假设输入数据已经排好序。如若不然，就要先对输入数据进行一次排序。

***，如果使用LINQ的话，还可以更简单：

 
 
 

  
  
  
IEnumerable, int, double>> Linq(IEnumerable, double>> tuples)  
  
  
  
{  
  
  
  
  var result = new List, int, double>>();  
  
  
  
  var q = from k in tuples group k by k.Item1;  
  
  
  
  foreach (var g in q) result.Add(new Tuple, int, double>(g.Key, g.Count(), g.Average(v => v.Item2)));  
  
  
  
  return result;  
  
  
  
} 
 
 
 

要注意LINQ 方法无论是运行时间还是占用的内存都更大。

我们来看看 Main 方法：

 
 
 

  
  
  
static void Main(string[] args)  
  
  
  
{  
  
  
  
  try  
  
  
  
  {  
  
  
  
    new Program().Run(Console.Out, int.Parse(args[0]));  
  
  
  
  }  
  
  
  
  catch (Exception ex)  
  
  
  
  {  
  
  
  
    Console.WriteLine(ex);  
  
  
  
  }  
  
  
  
}  
  
  
  
 
  
  
  
void Run(TextWriter writer, int n)  
  
  
  
{  
  
  
  
  var tuples = GetTuples(n * 1024 * 1024);  
  
  
  
  Write("ForEach", writer, ForEach(tuples));  
  
  
  
  Write("Iterate", writer, Iterate(tuples));  
  
  
  
  Write(" Linq  ", writer, Linq(tuples));  
  
  
  
} 
 
 
 

其中的 Write 方法如下所示：

 
 
 

  
  
  
void Write(string title, TextWriter writer, IEnumerable, int, double>> tuples)  
  
  
  
{  
  
  
  
  writer.WriteLine("==========> " + title + " <============");  
  
  
  
  writer.WriteLine("Key ------Count Average----------");  
  
  
  
  var count = 0;  
  
  
  
  var sum = 0.0;  
  
  
  
  foreach (var t in tuples)  
  
  
  
  {  
  
  
  
    writer.WriteLine("{0,3} {1,11:N0} {2}", t.Item1, t.Item2, t.Item3);  
  
  
  
    count += t.Item2;  
  
  
  
    sum += t.Item2 * t.Item3;  
  
  
  
  }  
  
  
  
  writer.WriteLine("--- ----------- -----------------");  
  
  
  
  writer.WriteLine("{0,3} {1,11:N0} {2}", tuples.Count(), count, sum / count);  
  
  
  
  writer.WriteLine();  
  
  
  
} 
 
 
 

***，这个程序的输出如下所示：

==========> ForEach <============
Key ------Count Average----------
  1      10,476 0.492122426354162
  2   1,633,289 0.499917991099794
  3     981,345 0.500446307804579
  5   1,542,377 0.500567888024527
  6     478,158 0.499376479287702
  8      62,325 0.501552373474687
  9   1,463,104 0.500270067230854
 11     802,680 0.500518684820775
 13     367,798 0.499572390413821
 14     492,947 0.500767958524
 16   2,403,053 0.500023199420802
 17     248,208 0.499988049057847
--- ----------- -----------------
 12  10,485,760 0.50018897689056

==========> Iterate <============
Key ------Count Average----------
  1      10,476 0.492122426354162
  2   1,633,289 0.499917991099794
  3     981,345 0.500446307804579
  5   1,542,377 0.500567888024527
  6     478,158 0.499376479287702
  8      62,325 0.501552373474687
  9   1,463,104 0.500270067230854
 11     802,680 0.500518684820775
 13     367,798 0.499572390413821
 14     492,947 0.500767958524
 16   2,403,053 0.500023199420802
 17     248,208 0.499988049057847
--- ----------- -----------------
 12  10,485,760 0.50018897689056

==========>  Linq   <============
Key ------Count Average----------
  1      10,476 0.492122426354162
  2   1,633,289 0.499917991099794
  3     981,345 0.500446307804579
  5   1,542,377 0.500567888024527
  6     478,158 0.499376479287702
  8      62,325 0.501552373474687
  9   1,463,104 0.500270067230854
 11     802,680 0.500518684820775
 13     367,798 0.499572390413821
 14     492,947 0.500767958524
 16   2,403,053 0.500023199420802
 17     248,208 0.499988049057847
--- ----------- -----------------
 12  10,485,760 0.50018897689056

 
这个程序中用到的 Tuple 类如下所示：

 
 
 

  
  
  
class Tuple, T2> 
  
  
  
{  
  
  
  
  public T1 Item1 { get; private set; }  
  
  
  
  public T2 Item2 { get; private set; }  
  
  
  
  public Tuple(T1 item1, T2 item2) { Item1 = item1; Item2 = item2; }  
  
  
  
}  
  
  
  
 
  
  
  
class Tuple, T2, T3> : Tuple, T2> 
  
  
  
{  
  
  
  
  public T3 Item3 { get; private set; }  
  
  
  
  public Tuple(T1 item1, T2 item2, T3 item3) : base(item1, item2) { Item3 = item3;  }  
  
  
  
} 
 
 
 

其实 .NET Framework 4.0 Base Class Library 中已经有 Tuple 类了。

本文中的全部源程序代码可以在这里下载。