如何通过解析CSV文件创建Map[K，V]和Seq[A]

Question

我对scala很陌生，我想通过解析csv文件来创建scalaMap()和scalaSeq()。

Input1.csv
-------------
State,Capital,Filter,Unique
MH,Mumbai,1,Yes
KA,Bengaluru,0,Yes
AP,Hydrabad,1,No
TS,Hydrabad,1,No

从Input1.csv文件我想要scala.collection.Map[K, V]

Map("MH" -> "Mumbai", "KA" -> "Bengaluru", "AP" -> "Hydrabad", "TS" -> "Hydrabad")

Input2.csv
-----------
Columns_Names
State
Capital
Filter
Unique

从Input2.csv文件我想要scala.collection.Seq[A]

Seq("State", "Capital", "Filter", "Unique")

如何在scala中做到这一点？

Answer 1

您可以这样做：

import scala.io.Source
import scala.util.{Failure, Success, Try}
val source = Source.fromFile("Input1.csv")
val cityMap = Try(source
.getLines()
.toSeq
.tail
.map {
x =>
x.split(",") match {
case Array(code, capital, _, _) =>
code -> capital
}
}.toMap) match {
case Failure(exception) =>
exception.printStackTrace()
Map.empty[String, String]
case Success(value) =>
value
}
println(cityMap) // Map(MH -> Mumbai, KA -> Bengaluru, AP -> Hydrabad, TS -> Hydrabad)
source.close()