优雅处理Shapely多边形外边界坐标的Python类型提示问题

我太懂你这个需求了——既要让类型检查工具（比如Pyright、mypy）能精准识别坐标的二元组类型，提前拦截像data[0][2]这种越界访问的低级错误，又不想写一堆冗余的循环代码，还得兼顾运行时效率对吧？

先拆解下你遇到的核心问题：Shapely的polygon.exterior.coords转成list后，类型检查器会把它识别为list[tuple[float, ...]]（变长float元组的列表），而你需要的是list[tuple[float, float]]（固定二元组的列表）。因为Python的list是不变类型（invariant），宽泛的类型不能直接赋值给更严格的类型，所以才会出现那个类型提示错误。

单个Polygon的优雅解决方案 #

不用写循环，直接用typing.cast做类型断言就搞定了——这相当于告诉类型检查器：“我明确知道这个转成list后的结果是二元组的列表，放心用！”，而且运行时完全没有额外开销（cast在运行时是个空操作）：

from typing import cast
from shapely import Polygon

polygon = Polygon([(0, 0), (1, 0), (1, 1), (0, 1)])
# 用cast直接给类型检查器明确标注类型
data: list[tuple[float, float]] = cast(list[tuple[float, float]], list(polygon.exterior.coords))

# 现在类型检查器会正确提示越界错误：Index 2 is out of range for type tuple[float, float]
data[0][2]

兼容MultiPolygon的简化实现 #

针对你需要处理Polygon | MultiPolygon的场景，同样用类型断言替代嵌套循环，代码简洁又高效：

from typing import cast
from shapely import Polygon, MultiPolygon

def get_coords(geometry: Polygon | MultiPolygon) -> list[tuple[float, float]] | list[list[tuple[float, float]]]:
    if isinstance(geometry, Polygon):
        # 断言单个多边形的外边界坐标是二元组列表
        return cast(list[tuple[float, float]], list(geometry.exterior.coords))
    elif isinstance(geometry, MultiPolygon):
        # 对每个子多边形做同样的断言，再组合成二维列表
        return cast(list[list[tuple[float, float]]], 
                    [cast(list[tuple[float, float]], list(poly.exterior.coords)) for poly in geometry.geoms])
    else:
        raise NotImplementedError(f"Unhandled type: {type(geometry)}")

为什么这个方案更优？ #

1. 无运行时额外开销：比你原来的循环方法快得多，因为直接用list()转换Shapely的coords迭代器，没有多余的元组解包和重新打包操作；
2. 代码更简洁：去掉了冗余的循环逻辑，可读性更强；
3. 类型检查精准：既能让类型检查器提前拦截越界错误，又能解决类型不兼容的提示问题。

补充说明 #

如果你用的是Python 3.10及以上版本，typing.Union可以直接用|替代，代码会更清爽，就像上面示例里写的那样。

备注：内容来源于stack exchange，提问作者AJP