基于osgeo判断矢量文件是点、线还是面文件

# -*- coding: utf-8 -*-
"""Created on Tue Jun 22 09:02:18 2021
@author: DeepLearning"""

from osgeo import ogr


def judge_geom_type(_input_shapefile):
    '''判断一个矢量文件是面、线还得点，
    如果是面返回值为：Polygon
    如果是线返回时为：Line String：
    如果是点返回值为：Point'''
    _ds = ogr.Open(_input_shapefile)
    _lyr = _ds.GetLayer()
    _lyr_defn = _lyr.GetLayerDefn()
    _ds.Destroy()
    return ogr.GeometryTypeToName(_lyr_defn.GetGeomType())


if __name__ == '__main__':
    input_shp = r'C:\Users\DeepLearning\Desktop\新建文件夹 (2)\dis_bid=37_GuiZhou_AnLongXian.shp'
    print(judge_geom_type(input_shp))