对于GIS分析领域的开发工程师，遍历文件夹中的shp文件是一个常见的需求。Python提供了一种非常便捷的方法来实现这个功能。本文将从以下几个方面进行讲解:
一、`os`模块的使用
`os`模块是Python内置的与操作系统相关的模块。使用`os`模块的`walk`函数可以遍历指定目录及其所有子目录下的所有文件和文件夹。下面是一个遍历目录下所有文件的示例代码：
import os
def getAllFiles(path):
fileList = []
for root, dirs, files in os.walk(path):
for file in files:
fileList.append(os.path.join(root, file))
return fileList
path = "your/dir/path"
allFiles = getAllFiles(path)
print(allFiles)
在这段代码中，我们使用了`os.walk`函数，函数返回一个三元组`(root, dirs, files)`，其含义如下：
- `root`: str类型，表示当前遍历的文件夹路径
- `dirs`: list类型，当前文件夹下所有子目录的列表
- `files`: list类型，当前文件夹下所有文件的列表
遍历到文件后，我们可以通过判断文件的后缀名来筛选出shp文件。下面是一个筛选shp文件的示例代码：
def getShpFiles(path):
fileList = []
for root, dirs, files in os.walk(path):
for file in files:
if file.endswith(".shp"):
fileList.append(os.path.join(root, file))
return fileList
path = "your/dir/path"
shpFiles = getShpFiles(path)
print(shpFiles)
二、使用`glob`模块
`glob`模块提供了一种更加简洁的方法来获取符合指定模式的文件列表。下面是一个使用`glob`模块获取shp文件列表的示例代码：
import glob path = "your/dir/path/**/*.shp" shpFiles = glob.glob(path, recursive=True) print(shpFiles)
在上面的代码中，我们使用了`**`来表示递归匹配任意子目录。如果不需要递归匹配，可以将该部分省略。另外，`glob`函数的`recursive`参数用来控制是否递归匹配，默认为False。
三、使用`pathlib`模块
`pathlib`是Python 3.4引入的一个路径操作模块，提供了一种面向对象的路径操作方式。使用`pathlib`模块可以更加直观地操作路径和文件。下面是一个使用`pathlib`模块获取shp文件列表的示例代码：
import pathlib
path = pathlib.Path("your/dir/path")
shpFiles = list(path.glob("**/*.shp"))
print(shpFiles)
在这个代码中，我们使用了`Path`类来表示路径。`glob`函数返回一个生成器，我们可以使用`list`函数将其转变为列表。
四、总结
本文介绍了三种不同的方法来遍历文件夹中的shp文件，分别使用了`os`模块、`glob`模块和`pathlib`模块。不同的方法各有特点，开发者可以根据具体情况选择适合自己的方法。
原创文章，作者：HAXFI，如若转载，请注明出处：https://www.506064.com/n/373772.html
微信扫一扫 
支付宝扫一扫 