Cala中的Scala flatMap.toMap类似物

时间:2012-04-07 08:01:53

标签: c# scala

前几天我在C#中编写了一些代码来获取一个对象列表,每个对象都包含一个数组字段,并将其转换为一个映射(即字典),其中对象成为值,并且对象的数组中的每个元素领域成为关键。这可能没有多大意义,所以让我们展示一些代码。在Scala中,我可能会这样做:

// Data class
class MyClass(val keys:Array[String])

object MyClassTests {

  // Dummy method just to get some sample data
  def getMyClassList() = List(
    new MyClass(Array("one", "two", "three")),
    new MyClass(Array("four", "five", "six")))


  def test() = {
    val list = getMyClassList()
    val map1 = list.view.flatMap(mc => mc.keys.map(_ -> mc)).toMap
    // or, if you like for-comprehensions:
    val map2 = (for (mc <- list.view; k <- mc.keys) yield k -> mc).toMap
    map1 // or map2, same difference
  }
}

在REPL中运行它,给我们(添加了格式):

res0: scala.collection.immutable.Map[String,MyClass] = 
    Map(four -> MyClass@4ee31ef2, 
        three -> MyClass@69bc6271, 
        two -> MyClass@69bc6271, 
        six -> MyClass@4ee31ef2, 
        five -> MyClass@4ee31ef2, 
        one -> MyClass@69bc6271)

但是,我不想在Scala中执行此操作,我想在C#中执行此操作。下面给出了两种可能的解决方案,Test1()是一种非常强制性的解决方案,而Test2()就像我现场想象的那样近似。

所以我的问题是这个:忽略这个例子看似无聊的事实,Test2()实际上是最接近Scala代码的模拟,并且它是最简洁的方法C#?是否有更简洁的方法来完成相同的任务(在C#中)?

// Data class
class MyClass {
    public string[] Keys { get; set; }

}

class MyClassTests {
    // Dummy method just to get some sample data
    public static IList<MyClass> GetMyClassList() {
        return new List<MyClass> {
            new MyClass {
                Keys = new[] {"one", "two", "three"}
            },
            new MyClass {
                Keys = new[] {"four", "five", "six"}
            }
        };
    }

    public void Test1() {
        var list = GetMyClassList();
        var validTypes = new Dictionary<string, MyClass>();
        foreach (var h in list) {
            foreach (var key in h.Keys)
                validTypes.Add(key, h);
        }
    }

    public void Test2() {
        var list = GetMyClassList();
        var validPartTypes = list
            .SelectMany(mc => mc.Keys.Select(k => new KeyValuePair<string,MyClass>(k, mc)))
            .ToDictionary(x => x.Key, x => x.Value);
    }
}

2 个答案:

答案 0 :(得分:4)

您可以使用匿名类型而不是KeyValuePair<>来使其更加简洁:

var validPartTypes = list.SelectMany(mc => mc.Keys.Select(k => new { k, mc }))
                         .ToDictionary(x => x.k, x => x.mc);

或者,如果您觉得LINQ查询语法更清晰,那就是:

var validPartTypes = (from mc in list
                      from k in mc.Keys
                      select new { k, mc })
                      .ToDictionary(x => x.k, x => x.mc);

但是你的代码完全有效,我认为在C#中没有更好的方法来实现它。

答案 1 :(得分:0)

您可以创建自己的LINQ方法:

public static class MyLinq
{
    public static IDictionary<TKey, TValue> ToDictionary<TKey, TValue>(this IEnumerable<KeyValuePair<TKey, TValue>> t)
    {
        return t.ToDictionary(p => p.Key, p => p.Value);
    }
}

而且你也可以使用ToDictionary而不是.Select(k => new KeyValuePair,但由于创建了许多无用的词典,这是一种非常缓慢的方式:

public void Test2() { 
    var list = GetMyClassList(); 
    var validPartTypes = list 
        .SelectMany(mc => mc.Keys.ToDictionary(k => mc)) 
        .ToDictionary(); 
}