Python 读取.shp文件并生成图幅编号的实现思路

import csv
import geopandas as gpd
def get_map_extent(shp_file_path):
    """
    读取.shp文件并获取其边界范围。
    :param shp_file_path: .shp文件路径
    :return: 地图边界的范围 (minx, miny, maxx, maxy)
    """
    # 读取.shp文件
    gdf = gpd.read_file(shp_file_path)
    # 获取地图边界
    boundary = gdf.geometry.total_bounds
    # 返回地图范围
    return boundary
def calculate_sheet_ranges(map_extent):
    """
    根据地图边界计算图幅范围，并按照南北纬和东西经的5度分隔规则返回图幅范围的格式。
    :param map_extent: 地图边界的范围 (minx, miny, maxx, maxy)
    :return: 图幅范围列表，每个元素包含(lon_start, lat_start, lon_end, lat_end, sheet_range)
    """
    minx, miny, maxx, maxy = map_extent
    # 初始化图幅范围列表
    sheet_ranges = []
    # 计算图幅编号
    # for lon_start in range(int(minx // 5) * 5, int(maxx // 5 + 1) * 5, 5):
    #     for lat_start in range(int(miny // 5) * 5, int(maxy // 5 + 1) * 5, 5):
    for lon_start in range(int(minx // 5) * 5, int(maxx // 5 + 1) * 5, 5):
        for lat_start in range(int(miny // 5 + 1) * 5, int(maxy // 5 + 2) * 5, 5):  # GLC_FCS30D 纬度需要上调一格5度
            lon_end = lon_start + 5
            lat_end = lat_start + 5
            sheet_range = f"{'W' if lon_start < 0 else 'E'}{abs(lon_start)}" \
                          f"{'S' if lat_start < 0 else 'N'}{abs(lat_start)}"
            sheet_ranges.append(sheet_range)
    return sheet_ranges
def main():
    iso_data = []
    with open('../../ISO.csv', newline='') as csvfile:
        # 创建 CSV 读取器
        csv_reader = csv.reader(csvfile)
        data = list(csv_reader)  # 将文件内容读取到内存中
        # 初始化行数计数器
        iso_count = 0
        for country in data:
            iso_count += 1
            sids_country = country[0]
            iso_map_extent = [iso_count, sids_country]  # [1, 'BMU']
            # 示例文件路径
            shp_file_path = fr'path_to_admin_division\{sids_country}.shp'
            # 获取地图范围
            map_extent = get_map_extent(shp_file_path=shp_file_path)
            # 计算图幅范围
            sheet_ranges = calculate_sheet_ranges(map_extent=map_extent)
            # 打印结果
            print(iso_count, sids_country, map_extent)
            iso_map_extent.extend(sheet_ranges)
            iso_data.append(iso_map_extent)
            print('- Map Extent Num: ', len(sheet_ranges), iso_map_extent[2:], '\n')
if __name__ == '__main__':
    main()