1 简介
在上一篇文章中我们对geopandas中的坐标参考系有了较为深入的学习，而在日常空间数据分析工作中矢量文件的读入和写出，是至关重要的环节。
作为基于geopandas的空间数据分析系列文章的第三篇，通过本文你将会学习到geopandas中的文件IO。
2 文件IO 2.1 矢量文件的读入
geopandas将fiona作为操纵矢量数据读写功能的后端。
使用geopandas.read_file()读取对应类型文件，而在后端实际上是使用fiona.open来读入数据，即两者参数是保持一致的，读入的数据自动转换为GeoDataFrame。
下面是geopandas.read_file()主要参数：
立即学习“Python免费学习笔记（深入）”；
下面结合上述参数，来介绍一下使用geopandas.read_file()在不同情况下读取常见格式矢量数据的方法。
使用到的示例数据为中国地图，CRS为EPSG:4326。
本文使用到的所有数据都可以在文章开头提及的Github仓库对应本文路径下找到：
图1
2.1.1 shapefile
作为非常常见的一种矢量文件格式，geopandas对shapefile提供了很好的读取和写出支持。
下面分为不同情况来介绍：
完整的shapefile
如图2，这是一个完整的shapefile：
图2
使用geopandas来读取这种形式的shapefile很简单：
代码语言：javascript代码运行次数：0运行复制
import geopandas as gpddata = gpd.read_file('geometry/china_provinces/china_provinces.shp')print(data.crs) # 查看数据对应的crsdata.head() # 查看前5行
图3
缺少投影的shapefile
当shapefile中缺失.prj文件时，使用geopandas读入后形成的GeoDataFrame会缺失crs属性：
图4
如果已经知道数据对应的CRS，可以在读入数据后补充上crs信息以进行其他操作：
代码语言：javascript代码运行次数：0运行复制
import pyprojdata.crs = pyproj.CRS.from_user_input('EPSG:4326')data.crs
图5
直接读取文件夹
当文件夹下只有单个shapefile时，可以直接读取该文件夹：
图6
读取zip压缩包中的文件
geopandas通过传入特定语法格式的文件路径信息，以支持直接读取.zip格式压缩包中的shapefile文件，主要分为两种情况。
当文件在压缩包内的根目录时，使用下面的语法规则来读取数据：
代码语言：javascript代码运行次数：0运行复制
zip://路径/xxx.zip譬如我们要读取图7所示的压缩包内文件：
图7
按照对应的语法规则，读取该类型数据方式如下：
图8
而当文件在压缩包内的文件夹中时，如图9：
图9
使用下面的语法规则来读取数据：
代码语言：javascript代码运行次数：0运行复制
zip://路径/xxx.zip!压缩包内指定文件路径将上述语法运用到上述文件：
图10
2.1.2 gdb与gpkg
对于Arcgis中的地理数据库gdb，以及QGIS中的GeoPackage，要读取其包含的矢量数据，就要涉及到图层的概念。
对应geopandas.read_file()的layer参数，只需要将gdb或gpkg文件路径作为filename参数，再将对应的图层名称作为layer参数传入：
gdb代码语言：javascript代码运行次数：0运行复制
data = gpd.read_file('geometry/china_provinces.gdb', layer='china_provinces')print(data.crs) # 查看数据对应的crsdata.head() # 查看前5行
图11
gpkg
类似读入gdb文件：
面试猫
AI面试助手，在线面试神器，助你轻松拿Offer
352 查看详情 代码语言：javascript代码运行次数：0运行复制
data = gpd.read_file('geometry/china_provinces.gpkg', layer='china_provinces', encoding='utf-8')print(data.crs) # 查看数据对应的crsdata.head() # 查看前5行
图12
2.1.3 GeoJSON
作为web地图中最常使用的矢量数据格式，GeoJSON几乎被所有在线地图框架作为数据源格式，在geopandas中读取GeoJSON非常简单，只需要传入文件路径名称即可。
下面我们来读入图13所示的文件：
图13
图14
2.1.4 过滤
geopandas在0.1.0版本中新增了bbox过滤，在0.7.0版本中新增了蒙版过滤和行过滤功能，可以辅助我们根据自己的需要读入原始数据中的子集。
下面一一进行介绍：
bbox过滤
bbox过滤允许我们在read_file()中传入一个边界框作为参数bbox，格式为(左下角x, 左下角y, 右上角x, 右上角y)，这样在读入的过程中只会保留几何对象与bbox有相交的数据记录。
下面我们仍然以上文中使用过的中国地图数据为例，我们在读入的过程中，传入边界框：
代码语言：javascript代码运行次数：0运行复制
from shapely import geometrydata = gpd.read_file('geometry/china_provinces.json', bbox=(100, 20, 110, 30))%matplotlib widgetax = data.plot()# 绘制bbox框示意ax = gpd.GeoSeries([geometry.box(minx=100, miny=20, maxx=110, maxy=30).boundary]).plot(ax=ax, color='red')
图15
可以看到只有跟红色框有相交的几何对象被读入。
蒙版过滤
蒙版过滤和bbox过滤功能相似，都是筛选与指定区域相交的数据记录。
不同的是蒙版过滤通过mask参数可以传入任意形状的多边形，不再像bbox过滤那样只接受矩形：
代码语言：javascript代码运行次数：0运行复制
data = gpd.read_file('geometry/china_provinces.json', mask=geometry.Polygon([(100, 20), (110, 30), (120, 20)]))ax = data.plot()# 绘制bbox框示意ax = gpd.GeoSeries([geometry.Polygon([(100, 20), (110, 30), (120, 20)]).boundary]).plot(ax=ax, color='red')
图16
可以看到只有跟红色多边形相交的几何对象被读入。
行过滤
行过滤的功能就比较简单，通过参数rows控制读入原数据的前若干行，可以用于在读取大型数据时先快速查看前几行以了解整个数据的格式：
图17
2.2 矢量文件的写出
在geopandas中使用to_file()来将GeoDataFrame或GeoSeries写出为矢量文件，主要支持shapefile、GeoJSON以及GeoPackage。
不像geopandas.read_file()可以根据传入的文件名称信息自动推断类型，我们在写出矢量数据时就需要使用driver参数来声明文件类型：
ESRI Shapefile
我们将上文最后一次读入的GeoDataFrame写出为ESRI Shapefile，设置driver参数为ESRI Shapefile。
如果你对文件编码有要求，这里可以使用encoding参数来指定，譬如这里我们指定为utf-8：
代码语言：javascript代码运行次数：0运行复制
'''在工程根目录下创建output文件夹'''import ostry: os.mkdir('output')except FileExistsError: pass data.to_file('output/output.shp', driver='ESRI Shapefile', encoding='utf-8')可以看到在output文件夹下，成功导出了完整的shapefile：
图18
而如果导出的文件名不加后缀扩展名，则会生成包含在新目录下的shapefile：
代码语言：javascript代码运行次数：0运行复制
data.to_file('output/output_shapefile', driver='ESRI Shapefile', encoding='utf-8')
图19
也可以向指定的文件夹下追加图层：
代码语言：javascript代码运行次数：0运行复制
data.to_file('output/output_shapefile_multi_layer', driver='ESRI Shapefile', layer='layer1', encoding='utf-8')data.to_file('output/output_shapefile_multi_layer', driver='ESRI Shapefile', layer='layer2', encoding='utf-8')data.to_file('output/output_shapefile_multi_layer', driver='ESRI Shapefile', layer='layer3', encoding='utf-8')
图20
GeoPackage
对于gdb文件，由于ESRI的限制，暂时无法在开源的geopandas中导出。
但我们可以用QGIS中的GeoPackage作为替代方案（开源世界万岁O(∩_∩)O~~），只需要将driver参数设置为GPKG即可。
这里需要注意一个bug：在使用geopandas导出GeoPackage文件时，可能会出现图21所示错误：
图21
但我观察到即使出现了上述错误，GeoPackage文件也是成功保存到路径下的且整个程序并未被打断，因此可以无视上述错误：
图22
GeoJSON
写出为GeoJSON非常容易，只需要设置driver='GeoJSON'即可：
图23
以上就是Python地信专题 | 基于geopandas的空间数据分析-文件IO篇的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/814337.html
微信扫一扫 
支付宝扫一扫 
