C# LINQ - 按属性对列表进行分组，然后按不同的组进行选择

Question

我有一个列表，我想按属性 A 进行分组。然后选择 GroupBy 键为 null 或 '' 的该组的所有元素，然后我想返回这些组的所有元素。否则，对于组键，返回该组的第一个元素。我想出了以下内容以及我对代码的期望。任何帮助将不胜感激。

示例代码:

public class ABC
{
    public string A { get; set; }
    public string B { get; set; }
    public string C { get; set; }
}

var list = new List<ABC>()
{
    new ABC() { A = "a", B = "b", C = "c" },
    new ABC() { A = "a", B = "b", C = "c" },
    new ABC() { A = null, B = "b", C = "c" },
    new ABC() { A = null, B = "b", C = "c" },
    new ABC() { A = "", B = "b", C = "c" },
    new ABC() { A = "", B = "b", C = "c" },
};

var result = list.GroupBy(x => x.A)
                 .Select(x => 
                 {
                     if (!string.IsNullOrEmpty(x.Key))
                        return x.First();
                     else
                        return x;
                 })
                 .ToList();

我的期望:

我希望从结果中看到总计 5

{ A = "a", B = "b", C = "c" } count 1 (1 removed)
{ A = null, B = "b", C = "c" } count 2 (none removed)
{ A = "", B = "b", C = "c" } count 2 (none removed)

Answer 1

像这样:

 var result = list
   .GroupBy(item => item.A)
   .SelectMany(group => string.IsNullOrEmpty(group.Key)
      ? group as IEnumerable<ABC>
      : new ABC[] {group.First()})
   .ToList();

这里的技巧是我们应该返回集合:整个组或仅包含First 项的集合(数组)。

您可以摆脱分组和扁平化以获得更快的版本，但是，它利用了副作用:

 HashSet<string> uniqueA = new HashSet<string>(); 

 var result = list
   .Where(item => string.IsNullOrEmpty(item.A) || uniqueA.Add(item.A))
   .ToList();