可能我的问题很明显。想要查看一个目录并创建字符串列表,其中每个string表示存储在给定目录中的文件名,例如,列表("file1.csv","file2.csv","file3.csv")。
我使用创建列表的function,但它是文件的list(不是字符串)并包括完整路径(不仅是文件名)。
import java.io.File
def getFileNames(path: String): List[File] = {
val d = new File(path)
if (d.exists && d.isDirectory) {
d
.listFiles // create list of File
.filter(_.isFile)
.toList
.sortBy(_.getAbsolutePath().replaceAll("[^a-zA-Z0-9]",""))
} else {
Nil // return empty list
}
}
谢谢你的所有想法。
分析解答
尝试将getFileNames
的返回类型更改为List[String]
并像这样使用map(_.getName)
def getFileNames(path: String): List[String] = {
val d = new File(path)
if (d.exists && d.isDirectory) {
d
.listFiles // create list of File
.filter(_.isFile)
.toList
.sortBy(_.getAbsolutePath().replaceAll("[^a-zA-Z0-9]",""))
.map(_.getName)
} else {
Nil // return empty list
}
}
确保.map(_.getName)
是链中的最后一个,即在sortBy
之后。
better-files会将此简化为
import better.files._
import better.files.Dsl._
val file = file"."
ls(file).toList.filter(_.isRegularFile).map(_.name)