学习Scala脚本：从文件里读取行记录-python 按行读取文件

处理琐碎的，每日工作的脚本经常需要处理文件。本节中，你将建立一个从文件中读入行记录，并把行中字符个数前置到每一行，打印输出的脚本。***版展示在代码3.10中：

import scala.io.Source  
 
if (args.length > 0) {  
 
 for (line <- Source.fromFile(args(0)).getLines)  
  print(line.length + " " + line)  
}  
else  
 Console.err.println("Please enter filename")

代码 3.10 从文件中读入行

此脚本开始于从包scala.io引用名为Source的类。然后检查是否命令行里定义了至少一个参数。若是，则***个参数被解释为要打开和处理的文件名。表达式Source.fromFile(args(0))，尝试打开指定的文件并返回一个Source对象，你在其上调用getLines。函数返回Iterator[String]，在每个枚举里提供一行包括行结束符的信息。for表达式枚举这些行并打印每行的长度，空格和这行记录。如果命令行里没有提供参数，***的else子句将在标准错误流中打印一条信息。如果你把这些代码放在文件contchars1.scala，并运行它调用自己：

$ scala countchars1.scala countchars1.scala

你会看到：

23 import scala.io.Source  
1  
23 if (args.length > 0) {  
1  
50   for (line <- Source.fromFile(args(0)).getLines)  
36     print(line.length + " " + line)  
2 }  
5 else  
47 Console.err.println("Please enter filename")

尽管当前形式的脚本打印出了所需的信息，你或许希望能让数字右序排列，并加上管道符号，这样输出看上去就替换成：

23 | import scala.io.Source  
 1 |  
23 | if (args.length > 0) {  
 1 |  
50 |   for (line <- Source.fromFile(args(0)).getLines)  
36 |     print(line.length + " " + line)  
 2 | }  
 5 | else  
47 |   Console.err.println("Please enter filename")

想要达到这一点，你可以对所有行枚举两次。***次决定每行字符计数的***宽度。第二次打印输出之前计算的***宽度。因为要枚举两次，你***把它们赋给变量：

val lines = Source.fromFile(args(0)).getLines.toList

***的toList是必须加的，因为getLines方法返回的是枚举器。一旦你使用它完成遍历，枚举器就失效了。而通过调用toList把它转换为List，你就可以枚举任意次数，代价就是把文件中的所有行一次性贮存在内存里。lines变量因此就指向着包含了命令行指定的文件文本字串的数组。

下一步，因为要对每行字符数计算两次，每个枚举计算一次，你或许会考虑把表达式拉出来变成一个小函数，专门用来计算传入字串的字符长度：

def widthOfLength(s: String) = s.length.toString.length

有了这个函数，你就可以计算***长度了：

var maxWidth = 0 
for (line <- lines)  
 maxWidthmaxWidth = maxWidth.max(widthOfLength(line))

这里你用一个for表达式枚举了每一行，计算这些行的宽度，并且，如果比当前***宽度还大，就把它赋值给maxWidth，一个初始化为0的var。（max方法是你可以在任何Int上调用的，可以返回被调用者和被传入者中的较大的值。）如果你希望不用var发现***值，替代的方法是可以首先找到最长的一行，如：

val longestLine = lines.reduceLeft(  
 (a, b) => if (a.length > b.length) a else b  
)  
 
val widths = lines.map(widthOfLength)

reduceLeft方法把传入的方法应用于lines的前两个元素，然后再应用于***次应用的结果和lines接下去的一个元素，等等，直至整个列表。每次这样的应用，结果将是碰到的最长一行，因为传入的函数，(a, b) => if (a.length > b.length) a else b，返回两个传入字串的最长那个。reduceLeft将传回***一次应用的结果，也就是本例lines中包含的最长字串。

得到这个结果之后，你可以通过把最长一行传给widthOfLength计算***的宽度：

val maxWidth = widthOfLength(longestLine)

***剩下的就是用一个合适的格式把这些行打印出来。你可以这么做：

for (line <- lines) {  
 val numSpaces = maxWidth - widthOfLength(line)  
 val padding = " " * numSpaces  
 print(padding + line.length + " | " + line)  
}

在这个for表达式里，你再一次枚举了全部行记录。对于每一行，首先计算行长度前所需的空格并把它赋给numSpaces。然后用表达式：" " * numSpaces创建包含numSpaces个空格的字串。最终，你打印出你想要格式的信息。全部的脚本展示在代码3.11中：

import scala.io.Source  
def widthOfLength(s: String) = s.length.toString.length  
if (args.length > 0) {  
 val lines = Source.fromFile(args(0)).getLines.toList  
 val longestLine = lines.reduceLeft(  
  (a, b) => if (a.length > b.length) a else b  
 )  
 val maxWidth = widthOfLength(longestLine)  
 for (line <- lines) {  
  val numSpaces = maxWidth widthOfLength(line)  
  val padding = " " * numSpaces  
  print(padding + line.length +" | "+ line)  
 }  
}  
else  
 Console.err.println("Please enter filename")

代码 3.11 对文件的每行记录打印格式化的字符数量。

本文节选自《Programming in Scala》

【相关阅读】