大数据学习路线分享Scala系列之集合操作函数

126 阅读3分钟

大数据学习路线继续为大家分享Scala系列之集合操作函数

4.6 集合的重要函数

4.6.1sum/max/min/count

在序列中查找最大或最小值是一个极常见的需求,如下:

val numbers = Seq(11, 2, 5, 1, 6, 3, 9)
numbers.max

//11

numbers.min
//1

更高级的例子,其中包含一个书的序列

case class Book(title: String, pages: Int)
val books = Seq( Book("Future of Scala developers", 85),
Book("Parallel algorithms", 240),
Book("Object Oriented Programming", 130),
Book("Mobile Development", 495) )

//Book(Mobile Development,495)

books.maxBy(book => book.pages)
//Book(Future of Scala developers,85)

books.minBy(book => book.pages)

如上所示,minBy & maxBy 方法解决了复杂数据的问题。你只需选择决定数据最大或最小的属性。

4.6.2 过滤

过滤一个数字 List,只获取奇数的元素。

val numbers = Seq(1,2,3,4,5,6,7,8,9,10) numbers.filter(n => n % 2 == 0)

val books = Seq( Book("Future of Scala developers", 85),
Book("Parallel algorithms", 240),
Book("Object Oriented Programming", 130),
Book("Mobile Development", 495) )
books.filter(book => book.pages >= 120)

4.6.3 Flatten

val abcd = Seq('a', 'b', 'c', 'd')
val efgj = Seq('e', 'f', 'g', 'h')
val ijkl = Seq('i', 'j', 'k', 'l')
val mnop = Seq('m', 'n', 'o', 'p')
val qrst = Seq('q', 'r', 's', 't')
val uvwx = Seq('u', 'v', 'w', 'x')
val yz = Seq('y', 'z')
val alphabet = Seq(abcd, efgj, ijkl, mnop, qrst, uvwx, yz)
//
// List(a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t,
// u, v, w, x, y, z)
alphabet.flatten

当有一个集合的集合,然后你想对这些集合的所有元素进行操作时,就会用到 flatten。

4.6.4集合之间的操作

差集、交集和并集

val num1 = Seq(1, 2, 3, 4, 5, 6)
val num2 = Seq(4, 5, 6, 7, 8, 9)

//List(1, 2, 3)

num1.diff(num2)
//List(4, 5, 6)

num1.intersect(num2)
//List(1, 2, 3, 4, 5, 6, 4, 5, 6, 7, 8, 9)

num1.union(num2)

上述示例中的 union 保留了重复的元素。如果我们不需要重复怎么办?这时可以使用 distinct 函数

//List(1, 2, 3, 4, 5, 6, 7, 8, 9)
num1.union(num2).distinct

下面是上述功能的图示:

4.6.5 map(映射)列表元素

map 是 Scala 集合最常用的一个函数。它的功能十分强大:

val numbers = Seq(1,2,3,4,5,6)

//List(2, 4, 6, 8, 10, 12)

numbers.map(n => n * 2)
val chars = Seq('a', 'b', 'c', 'd')
//List(A, B, C, D)

chars.map(ch => ch.toUpper)

map 函数的逻辑是遍历集合中的元素并对每个元素调用函数。

4.6.6 flatMap

flatMap 是由下列这两个函数组成的:

map & flatten

例子:

val abcd = Seq('a', 'b', 'c', 'd')

//List(A, a, B, b, C, c, D, d)

abcd.flatMap(ch => List(ch.toUpper, ch))

4.6.7 对整个集合进行条件检查

1. val numbers = Seq(3, 7, 2, 9, 6, 5, 1, 4, 2)
2. //ture numbers.forall(n => n < 10)
3. //false numbers.forall(n => n > 5)

而 forall 函数就是为处理这类需求而创建的。

4.6.8 对集合进行分组

比如把某个集合拆分成偶数集和奇数集,partition 函数可以帮我们做到这一点:

val numbers = Seq(3, 7, 2, 9, 6, 5, 1, 4, 2)
//(List(2, 6, 4, 2), List(3, 7, 9, 5, 1))
numbers.partition(n => n % 2 == 0)

4.6.9 Fold

另一个流行的操作是 fold,通常可以考虑 foldLeft 和 foldRight。他们是从不同的方面做同样的工作:

val numbers = Seq(1, 2, 3, 4, 5)

//15 numbers.foldLeft(0)((res, n) => res + n)

val words = Seq("apple", "dog", "table")
//13 words.foldLeft(0)((resultLength, word) => resultLength + word.length)

foldLeft, reduceRight, and foldRight

方法foldLeft与reduceLeft工作方法很象,但是它让你指定一个值作为第一个元素。

scala> val a = Array(1, 2, 3)
a: Array[Int] = Array(1, 2, 3)

scala> a.reduceLeft(_+_)
res6: Int = 6

scala> a.foldLeft(100)(_+_)
res7: Int = 106

scala> a.foldLeft(200)(_+_)
res8: Int = 206