如何从 csv 字符串获取地图

Question

我对 Scala 相当陌生，但我现在正在做练习。
我有一个像这样的字符串

"A>Augsburg;B>Berlin"

. What I want at the end is a map

val mymap = Map("A"->"Augsburg", "B"->"Berlin")

我所做的是：

val st = locations.split(";").map(dynamicListExtract _)

with the function

private def dynamicListExtract(input: String)  = {
    if (input contains ">") {
      val split = input split ">"
      Some(split(0), split(1)) // return key , value
    } else {
      None 
    } 
  }

Now I have an

Array[Option[(String, String)

How do I elegantly convert this into a Map[String, String]

有人可以帮忙吗？ 谢谢

Answer 1

只需将 map 调用更改为 flatMap：

scala> sPairs.split(";").flatMap(dynamicListExtract _)
res1: Array[(java.lang.String, java.lang.String)] = Array((A,Augsburg), (B,Berlin))

scala> Map(sPairs.split(";").flatMap(dynamicListExtract _): _*)
res2: scala.collection.immutable.Map[java.lang.String,java.lang.String] = Map((A,Augsburg), (B,Berlin))

作为比较：

scala> Map("A" -> "Augsburg", "B" -> "Berlin")
res3: scala.collection.immutable.Map[java.lang.String,java.lang.String] = Map((A,Augsburg), (B,Berlin))

Answer 2

在 2.8 中，您可以执行以下操作：

val locations = "A>Augsburg;B>Berlin"
val result = locations.split(";").map(_ split ">") collect { case Array(k, v) => (k, v) } toMap

collect 类似于 map，但也会过滤部分函数中未定义的值。 toMap 将从 Traversable 创建一个 Map，只要它是 Traversable[(K, V)]。

Answer 3

Randall 的 for-compression 形式的解决方案也值得一看，它可能更清晰，或者至少让您更好地了解 flatMap 正在做什么。

Map.empty ++ (for(possiblePair<-sPairs.split(";"); pair<-dynamicListExtract(possiblePair)) yield pair)

Answer 4

一个简单的解决方案（不处理错误情况）：

val str = "A>Aus;B>Ber"
var map = Map[String,String]()

str.split(";").map(_.split(">")).foreach(a=>map += a(0) -> a(1))

但 Ben Lings 的更好。

Answer 5

val str= "A>Augsburg;B>Berlin"

Map(str.split(";").map(_ split ">").map(s => (s(0),s(1))):_*)

--or--

str.split(";").map(_ split ">").foldLeft(Map[String,String]())((m,s) => m + (s(0) -> s(1)))