如何获取matplotlib contour绘制的特定等高线的(x,y)坐标？

要提取contour绘制的特定z值等高线坐标，核心是利用contour函数返回的ContourSet对象——它存储了所有等高线的路径与数值信息。下面是具体实现步骤，先修正你示例代码里的小问题（原代码的z是一维数组，contour需要二维网格数据），再演示提取逻辑：

1. 修正基础代码并获取ContourSet对象 #

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(20,40,100)
y = np.linspace(0,100,100)
# 生成二维网格矩阵，这是contour要求的输入格式
X, Y = np.meshgrid(x, y)
Z = X**2 + Y**2  # 现在Z是与X/Y匹配的二维数组

plt.figure()
cs = plt.contour(X, Y, Z)  # 接收返回的ContourSet对象，后续从这里提取数据
plt.show()

2. 提取目标z值的等高线坐标 #

假设你要提取z=5000的等高线坐标，直接遍历ContourSet中的数据即可（注意浮点数值匹配要用近似判断，避免精度问题）：

target_z = 5000

# 遍历所有等高线集合与对应的z值
for idx, contour in enumerate(cs.collections):
    current_z = cs.levels[idx]
    # 用np.isclose处理浮点数值的近似匹配
    if np.isclose(current_z, target_z):
        # 一条等高线可能包含多段不连续路径，遍历所有路径
        for path in contour.get_paths():
            # path.vertices就是该段路径的(x,y)坐标数组，shape为(n,2)
            coords = path.vertices
            x_coords = coords[:, 0]
            y_coords = coords[:, 1]
            
            # 这里可以打印、保存或使用坐标数据
            print(f"z={current_z}的等高线坐标：")
            print("x坐标片段：", x_coords[:5])  # 只打印前5个示例
            print("y坐标片段：", y_coords[:5])

额外技巧：指定绘制目标等高线 #

如果你只需要某一条特定z值的等高线，可以在调用contour时直接传入levels参数，这样后续提取更简单：

# 只绘制z=5000的等高线
cs = plt.contour(X, Y, Z, levels=[5000])
# 直接提取这条唯一的等高线坐标
path = cs.collections[0].get_paths()[0]
coords = path.vertices
x_coords = coords[:, 0]
y_coords = coords[:, 1]

内容的提问来源于stack exchange，提问作者SR_00