scala - Scala：如何将模式匹配功能应用于包含以下值的 csv 文件

Question

def matchcase(x:String):Int =  x match{
        case "Iris-setosa" => 10
        case "Iris-virginica" => 20
        case "Iris-versicolor"=> 30
        case _ => 0

csv文件中的样本数据如下：

1,5.1,3.5,1.4,0.2,Iris-setosa
2,4.9,3,1.4,0.2,Iris-setosa
3,4.7,3.2,1.3,0.2,Iris-setosa
51,7,3.2,4.7,1.4,Iris-versicolor
52,6.4,3.2,4.5,1.5,Iris-versicolor
53,6.9,3.1,4.9,1.5,Iris-versicolor
103,7.1,3,5.9,2.1,Iris-virginica
104,6.3,2.9,5.6,1.8,Iris-virginica
105,6.5,3,5.8,2.2,Iris-virginica

标签： scalauser-defined-functions

Answer 1

在 Scala 中使用提取器模式和正则表达式

给定正则表达式

val regex = "[[0-9]|,|\\.]+([[a-zA-z]|\\-]+)".r

现在只需逐行模式匹配整个文本

lines.map {
  case regex(name) => Some(name)
  case _ => None
}

稍后使用您的matchCase函数将字符串（名称）转换为数字并将名称替换为数字

您正在寻找的代码

lines.map { line =>
  line match {
   case regex(name) => line.replace("[[a-zA-z]|\\-]+", matchcase(name).toString)
   case _ => line
  }
}

斯卡拉 REPL

scala> val regex = "[[0-9]|,|\\.]+([[a-zA-z]|\\-]+)".r
regex: scala.util.matching.Regex = [[0-9]|,|\.]+([[a-zA-z]|\-]+)

scala> "1,5.1,3.5,1.4,0.2,Iris-setosa" match { case regex(str) => println(s"name: $str")}
name: Iris-setosa

下一个

处理整个文本

scala> val text = """
     | 1,5.1,3.5,1.4,0.2,Iris-setosa
     | 2,4.9,3,1.4,0.2,Iris-setosa
     | 3,4.7,3.2,1.3,0.2,Iris-setosa
     | 51,7,3.2,4.7,1.4,Iris-versicolor
     | 52,6.4,3.2,4.5,1.5,Iris-versicolor
     | 53,6.9,3.1,4.9,1.5,Iris-versicolor
     | 103,7.1,3,5.9,2.1,Iris-virginica
     | 104,6.3,2.9,5.6,1.8,Iris-virginica
     | 105,6.5,3,5.8,2.2,Iris-virginica
     | """.stripMargin
text: String =
"
1,5.1,3.5,1.4,0.2,Iris-setosa
2,4.9,3,1.4,0.2,Iris-setosa
3,4.7,3.2,1.3,0.2,Iris-setosa
51,7,3.2,4.7,1.4,Iris-versicolor
52,6.4,3.2,4.5,1.5,Iris-versicolor
53,6.9,3.1,4.9,1.5,Iris-versicolor
103,7.1,3,5.9,2.1,Iris-virginica
104,6.3,2.9,5.6,1.8,Iris-virginica
105,6.5,3,5.8,2.2,Iris-virginica
"

scala> val lines = text.split("\n").filter(_.trim.nonEmpty)
lines: Array[String] = Array(1,5.1,3.5,1.4,0.2,Iris-setosa, 2,4.9,3,1.4,0.2,Iris-setosa, 3,4.7,3.2,1.3,0.2,Iris-setosa, 51,7,3.2,4.7,1.4,Iris-versicolor, 52,6.4,3.2,4.5,1.5,Iris-versicolor, 53,6.9,3.1,4.9,1.5,Iris-versicolor, 103,7.1,3,5.9,2.1,Iris-virginica, 104,6.3,2.9,5.6,1.8,Iris-virginica, 105,6.5,3,5.8,2.2,Iris-virginica)

scala> lines.map {
     |  case regex(name) => Some(name)
     |  case _ => None
     | }
res18: Array[Option[String]] = Array(Some(Iris-setosa), Some(Iris-setosa), Some(Iris-setosa), Some(Iris-versicolor), Some(Iris-versicolor), Some(Iris-versicolor), Some(Iris-virginica), Some(Iris-virginica), Some(Iris-virginica))

现在使用collect并且只收集以下值Some

scala> res18.collect { case Some(value) => value }
res19: Array[String] = Array(Iris-setosa, Iris-setosa, Iris-setosa, Iris-versicolor, Iris-versicolor, Iris-versicolor, Iris-virginica, Iris-virginica, Iris-virginica)

scala> res19.mkString("\n")
res20: String =
Iris-setosa
Iris-setosa
Iris-setosa
Iris-versicolor
Iris-versicolor
Iris-versicolor
Iris-virginica
Iris-virginica
Iris-virginica