学科分类
目录
Spark技术栈

数组

对于每一门编程语言来说,数组(Array)都是重要的数据结构之一,主要用来存储数据类型相同的元素。下面,针对数组的定义与使用、数组遍历以及数组转换操作进行详细介绍。

  1. 数组定义与使用

Scala中的数组分为定长数组和变长数组,这两种数组的定义方式如下:

new Array[T](数组长度) //定义定长数组

ArrayBuffer[T]()    //定义变长数组

上述语法格式中,定义定长数组,需要使用new关键字,而定义变长数组时,则需要导包import scala.collection.mutable.ArrayBuffer。[T]表示的是数组元素的类型,T为泛型。

当我们定义好数组后,可以对数组进行追加、插入以及删除等操作。针对不同的数组操作,Array提供了不同的API。

下面,我们通过一个例子来演示Scala数组简单使用,具体代码如文件1所示。

​ 文件1 ArrayDemo.scala

 1  import scala.collection.mutable.ArrayBuffer

 2  object ArrayDemo {

 3    def main(args: Array[String]) {

 4      //初始化一个长度为8的定长数组,其所有元素均为0

 5      val arr1 = new Array[Int](8)

 6      //直接打印定长数组,内容为数组的hashcode值

 7      println(arr1)

 8      //变长数组(数组缓冲)

 9      //如果使用数组缓冲,需要导入import scala.collection.mutable.ArrayBuffer

 10     val ab = ArrayBuffer[Int]()

 11     //向数组缓冲的尾部追加一个元素

 12     //+=尾部追加元素

 13     ab += 1

 14     println(ab)

 15     //追加多个元素

 16     ab += (2, 3, 4, 5)

 17     println(ab)

 18     //追加一个数组++=

 19     ab ++= Array(6, 7)

 20     println(ab)

 21      //追加一个数组缓冲

 22     ab ++= ArrayBuffer(8,9)

 23     println(ab)

 24     //打印数组缓冲ab

 25     //在数组某个位置插入元素用insert,从某下标插入

 26     ab.insert(0, -1, 0)

 27     println(ab)

 28     //删除数组某个位置的元素用remove按照下标删除

 29     ab.remove(0)

 30     println(ab)

 31   }

 32 }                                                           

上述代码中,第5-7行代码定义了一个定长数组arr1并打印数组对象;第10-30行代码定义了一个变长数组ab并对数组对象进行了追加、插入和删除等操作。

运行文件1中的代码,控制台输出结果如图1所示。

img

​ 图1 Scala数组定义与使用的输出结果

  1. 数组遍历

Scala中,如果想要获取数组中的每一个元素,则需要将数组进行遍历操作。数组的遍历有三种方式,分别是for循环遍历、while循环遍历以及do...while循环遍历。在这里,我们使用for循环对数组进行遍历操作。具体代码如文件2所示。

​ 文件2 ArrayTraversal.scala

 1  object ArrayTraversal {

 2    def main(args: Array[String]) {

 3      var myArr = Array(1.9, 2.9, 3.4, 3.5)

 4      // 打印输出所有的数组元素

 5      for (x <- myArr) {

 6        print(x + " ")

 7      }

 8      //打印换行

 9      println()

 10     // 计算数组所有元素的总和

 11     var total = 0.0;

 12     for (i <- 0 to (myArr.length - 1)) {

 13       total += myArr(i);

 14     }

 15     println("总和为 " + total);

 16     // 查找数组中的最大元素

 17     var max = myArr(0);

 18     for (i <- 1 to (myArr.length - 1)) {

 19       if (myArr(i) > max) max = myArr(i);

 20     }

 21     println("最大值为 " + max);

 22   }

 23 }

上述代码中,第3-7行代码定义了一个定长数组myArr并通过遍历打印该数组;第11-15行代码定义一个变量total并赋值为0.0,通过遍历计算数组所有元素的总和;第17-21行定义了一个变量max并赋值为数组myArr中的第一个元素,通过遍历查找出myArr数组中的最大元素。

运行文件2中的代码,控制台输出结果如图2所示。

img

​ 图2 遍历数组的控制台打印输出

  1. 数组转换

数组转换就是通过yield关键字将原始的数组进行转换,会产生一个新的数组,然而原始的数组保持不变。下面我们来演示数组的转换,定义一个数组,实现将偶数取出乘以10后生成一个新的数组,具体代码如文件3所示。

​ 文件3 ArrayYieldTest.scala

 1  object ArrayYieldTest {

 2    def main(args: Array[String]) {

 3      //定义一个数组

 4      val arr = Array(1, 2, 3, 4, 5, 6, 7, 8, 9)

 5      //将偶数取出乘以10后再生成一个新的数组

 6      val newArr = for (e <- arr if e % 2 == 0) yield e * 10

 7      println(newArr.toBuffer)

 8    }

 9  }

上述代码中,第4-7行代码定义了一个定长数组arr并通过求偶和算术操作,将数组arr转换成一个新数组newArr,最终打印newArr数组。

运行文件3中的代码,控制台输出结果如图3所示。

img

​ 图3 数组转换的控制台输出

点击此处
隐藏目录