使用Python获取两个Contour轮廓交点坐标时出现IndexError及TypeError问题求助

一、搞定IndexError: list index out of range的问题 #

你遇到这个错误的核心原因有两个：

1. 3D坐标轴下的contour使用逻辑偏差：当你在projection='3d'的3D轴上调用contour时，它默认会在z=0平面尝试绘制等高线，但如果你的函数在这个平面没有对应的等值线（或者你指定的level不在数据范围内），返回的contour对象就会没有任何collections，这时候访问collections[0]自然会触发索引越界。
2. 对levels参数的误解：你写的contour(...,1)里的1是让matplotlib自动生成1条等高线，但这条线的level不一定是你需要的（比如你实际要找的是z=0的曲线）。如果自动选择的level在你的数据里没有对应等值线，就会出现空的collections。

修复步骤： #

• 明确指定目标level：比如你要找的是两个函数等于0时的曲线交点，就直接用levels=[0]来指定绘制z=0的等高线，确保contour能生成有效的collections。
• 优先用2D坐标轴：找2D曲线的交点完全不需要3D轴，用普通2D轴能避免额外的复杂度；如果一定要用3D轴，必须确认指定的level存在于你的数据取值范围内。

修改后的可运行代码：

from shapely import geometry
import matplotlib.pyplot as plt
import numpy as np

def findIntersection(contour1, contour2):
    # 先做空值检查，避免索引越界
    if not contour1.collections or not contour2.collections:
        raise ValueError("其中一条或两条等高线不存在，请检查level是否在数据范围内")
    p1 = contour1.collections[0].get_paths()[0]
    v1 = p1.vertices
    p2 = contour2.collections[0].get_paths()[0]
    v2 = p2.vertices
    poly1 = geometry.LineString(v1)
    poly2 = geometry.LineString(v2)
    intersection = poly1.intersection(poly2)
    return intersection

# 改用2D坐标轴，更适合处理平面曲线交点
figtst2 = plt.figure(figsize=(18.5, 10.5))
ax2 = plt.axes()  # 去掉projection='3d'

# 增加采样点数量，让等高线更平滑，交点计算更准确
x = np.linspace(0, 21, 200)
y = np.linspace(0, 21, 200)
X, Y = np.meshgrid(x, y)

# 明确指定levels=[0]，绘制z=0的等高线
c1 = ax2.contour(X, Y, ((X - 5) * (Y - 5) - 25), levels=[0], colors='green', linewidths=3)
c2 = ax2.contour(X, Y, (X**2 + Y**2 - 400), levels=[0], colors='orange', linewidths=3)

ax2.set_title('Contour: Finding intersections of two curves')
ax2.set_aspect('equal')  # 保持坐标轴比例，让圆形等高线显示正常

# 获取并打印交点
intersection_example = findIntersection(c1, c2)
print("交点坐标：", list(intersection_example.coords))
plt.show()

二、解决TypeError: Input z must be a 2D array的问题 #

这个错误逻辑很清晰：contour函数要求输入的X、Y是二维网格数组（由np.meshgrid生成），对应的Z也必须是二维数组，这样才能一一对应网格上的每个点的数值。如果直接传入一维的x、y，计算出的Z会是一维的，完全不符合函数要求。

修复方案： #

永远先用np.meshgrid(x, y)把一维的x、y转换成二维的X、Y网格，再基于X、Y计算二维的Z数组——就像你第一份能正常运行的3D代码那样，示例如下：

x = np.linspace(0, 21, 200)
y = np.linspace(0, 21, 200)
X, Y = np.meshgrid(x, y)
# 基于二维网格计算二维Z数组
Z1 = (X - 5) * (Y - 5) - 25
Z2 = X**2 + Y**2 - 400

这样再把X、Y、Z传入contour，就不会触发TypeError了。

额外小贴士 #

• 用shapely计算交点时，采样点太少会导致等高线粗糙，可能出现交点计算不准甚至漏交点的情况，建议适当提高采样点数量（比如从20改成200）。
• 可以在findIntersection函数里增加空值检查，避免因level无效导致程序直接崩溃，同时给出明确的错误提示。

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