Scala Iterator dropWhile()方法示例|极客笔记

Scala Iterator dropWhile()方法示例

在Scala中，Iterator是常用的数据类型之一。在遍历集合元素时，我们经常需要根据一定的条件来选择合适的元素，Scala提供了许多方法来支持这种操作，其中dropWhile()方法就是一种能够满足我们需求的方法。本篇文章将为大家详细介绍Scala Iterator的dropWhile()方法并给出示例代码。

阅读更多：Scala 教程

Iterator的dropWhile()方法

Iterator的dropWhile()方法用于返回一个新的迭代器，跳过满足某个条件的前缀元素，返回剩余的迭代器。

dropWhile()方法的具体定义如下：

def dropWhile(p: (A) ⇒ Boolean): Iterator[A]

其中p是一个带一个参数的函数，表示过滤的条件。函数p的参数类型是迭代器中存储元素的类型A，返回值是一个Boolean类型的值，表示是否需要进行过滤。

Scala Iterator dropWhile()方法示例

我们接下来通过几个示例来更好地了解dropWhile()方法的使用，这里我们选取了一些语言的关键字和符号来构建一个一维的字符串数组，代码如下：

val keywords = Array("+", "-", "*", "/", "if", "else", "for", "while", "do", "val", "var", "class", "object")

接下来我们分别进行如下操作：

剔除关键字+、-、*、/，保留剩下的元素。
从for关键字开始保留剩下的元素。

示例一：

下面的代码展示如何剔除字符串数组中的关键字+、-、*、/。

val filteredKeywords = keywords.dropWhile(x => x.equals("+") || x.equals("-") || x.equals("*") || x.equals("/"))
println(filteredKeywords.mkString(","))

输出结果：

if,else,for,while,do,val,var,class,object

在上面的代码中，x表示迭代器中存储元素的类型A，filterdKeywords表示过滤后的字符串数组，mkString(“,”)方法表示将一个集合中的所有元素用逗号分隔开，并转换为字符串输出。

示例二：

下面的代码展示如何过滤掉for关键字之前的元素，并输出之后的元素。

val filteredKeywords = keywords.dropWhile(x => !x.equals("for"))
println(filteredKeywords.mkString(","))

输出结果：

for,while,do,val,var,class,object

在上面的代码中，x表示迭代器中存储元素的类型A，filterdKeywords表示过滤后的字符串数组，mkString(“,”)方法表示将一个集合中的所有元素用逗号分隔开，并转换为字符串输出。!x.equals(“for”)的含义是当x等于for时条件返回false，dropWhile会停止过滤并返回此时迭代器的所有元素。