scala系列(6)-- 函数

249 阅读5分钟

1 foreach

foreach主要 格式

foreach(f: (A) => Unit): Unit

参数    f: (A) ⇒ Unit    接收一个函数对象<br />函数的输入参数为集合的元素,返回值为空
返回值  Unit

示例

// 定义一个列表
scala> val a = List(1,2,3,4)
a: List[Int] = List(1, 2, 3, 4)

// 迭代打印
scala> a.foreach((x:Int)=>println(x))
scala> a.foreach(x=>println(x))
scala> a.foreach(println(_))

2 map

格式

def map[B](f: (A) ⇒ B): TraversableOnce[B]

| 泛型    | [B]                | 指定map方法最终返回的集合泛型                                |
| 参数    | f: (A) ⇒ B         | 传入一个函数对象<br />该函数接收一个类型A(要转换的列表元素),返回值为类型B |
| 返回值  | TraversableOnce[B] | B类型的集合                                                  |

示例

就是一个将一种数据类型转换为另外一种数据类型的过程。
scala> a.map(x=>x+1)
res4: List[Int] = List(2, 3, 4, 5)
// 简写
scala> a.map(_ + 1)

3 flatMap

格式

// 可以把flatMap,理解为先map,然后再flatten
def flatMap[B](f: (A) ⇒ GenTraversableOnce[B]): TraversableOnce[B]

| 泛型        | [B]                            | 最终要转换的集合元素类型                                     |
| 参数        | f: (A) ⇒ GenTraversableOnce[B] | 传入一个函数对象<br />函数的参数是集合的元素<br />函数的返回值是一个集合 |
| 返回值      | TraversableOnce[B]             | B类型的集合                                                  |

示例

1、先对其list进行map操作,在进行flatten扁平化处理
// 定义文本行列表
scala> val a = List("hadoop hive spark flink flume", "kudu hbase sqoop storm")
a: List[String] = List(hadoop hive spark flink flume, kudu hbase sqoop storm)

// 使用map将文本行转换为单词数组
scala> a.map(x=>x.split(" "))
res5: List[Array[String]] = List(Array(hadoop, hive, spark, flink, flume), Array(kudu, hbase, sqoop, storm))

// 扁平化,将数组中的
scala> a.map(x=>x.split(" ")).flatten
res6: List[String] = List(hadoop, hive, spark, flink, flume, kudu, hbase, sqoop, storm)

2、简化处理,一步到位
scala> a.flatMap(_.split(" "))
res7: List[String] = List(hadoop, hive, spark, flink, flume, kudu, hbase, sqoop, storm)

4 filter

格式

def filter(p: (A) ⇒ Boolean): TraversableOnce[A]

| 参数       | p: (A) ⇒ Boolean   | 传入一个函数对象<br />接收一个集合类型的参数<br />返回布尔类型,满足条件返回true, 不满足返回false |
| 返回值     | TraversableOnce[A] | 列表                                                         |
// 返回的结果是满足条件的结果
scala> List(1,2,3,4,5,6,7,8,9).filter(_ % 2 == 0)
res8: List[Int] = List(2, 4, 6, 8)

5 sorted、sortBy、sortWith

5.1sorted

参考代码

scala> List(3,1,2,9,7).sorted
res16: List[Int] = List(1, 2, 3, 7, 9)

5.2 sortBy


**方法签名**
def sortBy[B](f: (A) ⇒ B): List[A]

| 泛型       | [B]        | 按照什么类型来进行排序                                       |
| 参数       | f: (A) ⇒ B | 传入函数对象<br />接收一个集合类型的元素参数<br />返回B类型的元素进行排序 |
| 返回值     | List[A]    | 返回排序后的列表                                             |

参考代码

scala> val a = List("01 hadoop", "02 flume", "03 hive", "04 spark")
a: List[String] = List(01 hadoop, 02 flume, 03 hive, 04 spark)

// 获取单词字段
scala> a.sortBy(x => x.split(" ")(1))
// 简写
scala> a.sortBy(_.split(" ")(1))
res8: List[String] = List(02 flume, 01 hadoop, 03 hive, 04 spark)

5.3 sortWith

格式

def sortWith(lt: (A, A) ⇒ Boolean): List[A]

| 参数         | lt: (A, A) ⇒ Boolean | 传入一个比较大小的函数对象<br />接收两个集合类型的元素参数<br />返回两个元素大小,小于返回true,大于返回false |
| 返回值       | List[A]              | 返回排序后的列表                                             |

参考代码

scala> val a = List(2,3,1,6,4,5)
a: List[Int] = List(2, 3, 1, 6, 4, 5)

scala> a.sortWith((x,y) => if(x<y)true else false)
res15: List[Int] = List(1, 2, 3, 4, 5, 6)

scala> res15.reverse
res18: List[Int] = List(6, 5, 4, 3, 2, 1)

// 函数参数只在函数中出现一次,可以使用下划线代替
scala> a.sortWith(_ < _).reverse
res19: List[Int] = List(6, 5, 4, 3, 2, 1)

6 groupBy

格式

def groupBy[K](f: (A) ⇒ K): Map[K, List[A]]

| 泛型        | [K]             | 分组字段的类型                                               |
| 参数        | f: (A) ⇒ K      | 传入一个函数对象<br />接收集合元素类型的参数<br />返回一个K类型的key,这个key会用来进行分组,相同的key放在一组中 |
| 返回值      | Map[K, List[A]] | 返回一个映射,K为分组字段,List为这个分组字段对应的一组数据  |

参考代码

scala> val a = List("张三"->"男", "李四"->"女", "王五"->"男")
a: List[(String, String)] = List((张三,男), (李四,女), (王五,男))

// 按照性别分组
scala> a.groupBy(_._2)
res0: scala.collection.immutable.Map[String,List[(String, String)]] = Map(男 -> List((张三,男), (王五,男)),
女 -> List((李四,女)))

// 将分组后的映射转换为性别/人数元组列表
scala> res0.map(x => x._1 -> x._2.size)
res3: scala.collection.immutable.Map[String,Int] = Map(男 -> 2, 女 -> 1)

7 reduce

7.1 reduce

格式

def reduce[A1 >: A](op: (A1, A1) ⇒ A1): A1


| 泛型       | [A1 >: A]         | (下界)A1必须是集合元素类型的父类                           |
| 参数       | op: (A1, A1) ⇒ A1 | 传入函数对象,用来不断进行聚合操作<br />第一个A1类型参数为:当前聚合后的变量<br />第二个A1类型参数为:当前要进行聚合的元素 |
| 返回值     | A1                | 列表最终聚合为一个元素                                       |

参考代码

scala> val a = List(1,2,3,4,5,6,7,8,9,10)
a: List[Int] = List(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)

scala> a.reduce((x,y) => x + y)
res5: Int = 55

// 第一个下划线表示第一个参数,就是历史的聚合数据结果
// 第二个下划线表示第二个参数,就是当前要聚合的数据元素
scala> a.reduce(_ + _)
res53: Int = 55

// 与reduce一样,从左往右计算
scala> a.reduceLeft(_ + _)
res0: Int = 55

// 从右往左聚合计算
scala> a.reduceRight(_ + _)
res1: Int = 55

7.2 fold

格式

def fold[A1 >: A](z: A1)(op: (A1, A1) ⇒ A1): A1

| 泛型       | [A1 >: A]         | (下界)A1必须是集合元素类型的父类                           |
| 参数1      | z: A1             | 初始值                                                       |
| 参数2      | op: (A1, A1) ⇒ A1 | 传入函数对象,用来不断进行折叠操作<br />第一个A1类型参数为:当前折叠后的变量<br />第二个A1类型参数为:当前要进行折叠的元素 |
| 返回值     | A1                | 列表最终折叠为一个元素                                       |

参考代码

scala> val a = List(1,2,3,4,5,6,7,8,9,10)
a: List[Int] = List(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)

scala> a.fold(0)(_ + _)
res4: Int = 155