Scala迭代器indexWhere()方法及其示例

在Scala中，迭代器是一种非常常见的数据结构。迭代器对象可以在集合中进行元素遍历，在遍历过程中，我们可能需要查找某个指定元素的位置，这个时候，就可以使用迭代器的 indexWhere() 方法了。

阅读更多：Scala 教程

indexWhere()方法的介绍

Scaladoc描述了indexWhere(p: (A) => Boolean): Int用于查找满足条件的元素的位置。该方法将迭代器中的每个元素传递给给定的谓词函数，如果谓词函数返回true，则返回该元素的位置。

更通俗的说，就是在迭代器中查找满足特定条件的元素，并返回其位置。

indexWhere()方法的示例

我们通过一个简单的示例来介绍如何使用 indexWhere() 方法。

首先，我们需要创建一个List集合：

val lst = List(1,2,3,4,5,6)

然后，我们将lst集合转换为迭代器对象：

val iterator = lst.iterator

接下来，我们使用 indexWhere() 方法来查找元素2的位置：

val index = iterator.indexWhere(_ == 2)

最后，我们打印查找到的位置：

println(index) //1

完整代码如下：

val lst = List(1,2,3,4,5,6)
val iterator = lst.iterator
val index = iterator.indexWhere(_ == 2)
println(index) //1

indexWhere()方法的扩展示例

下面，我们再来看一个稍微复杂一些的示例。

首先，我们定义一个存储学生信息的List集合：

val students = List(
    Map("name" -> "Tom", "age" -> 20),
    Map("name" -> "Jack", "age" -> 22),
    Map("name" -> "Kate", "age" -> 18),
    Map("name" -> "Jerry", "age" -> 19))

然后，我们创建一个返回函数的方法，该方法将一个学生信息的Map作为参数，如果学生的年龄小于20，则返回true。

def isYoungerThan20(student: Map[String, Any]): Boolean = {
  student("age").asInstanceOf[Int] < 20
}

接下来，我们将students转为迭代器，并使用 indexWhere() 方法来查找第一个年龄小于20的学生：

val iterator = students.iterator
val index = iterator.indexWhere(isYoungerThan20)

最后，我们打印查找到的学生信息：

println(students(index)) //Map(name -> Kate, age -> 18)

完整代码如下：

val students = List(
    Map("name" -> "Tom", "age" -> 20),
    Map("name" -> "Jack", "age" -> 22),
    Map("name" -> "Kate", "age" -> 18),
    Map("name" -> "Jerry", "age" -> 19))

def isYoungerThan20(student: Map[String, Any]): Boolean = {
  student("age").asInstanceOf[Int] < 20
}

val iterator = students.iterator
val index = iterator.indexWhere(isYoungerThan20)
println(students(index)) //Map(name -> Kate, age -> 18)