如何绘制SymPy Geometry模块中的Segment2D对象？

你遇到的错误是因为SymPy的plot()函数默认不支持Geometry模块的Segment2D对象——它主要面向符号表达式，而非几何实体，直接传入会导致无法识别的迭代错误。

以下是两种可行的解决方法：

方法一：基于SymPy plot函数绘制 #

提取Segment2D的两个端点坐标，将其转为坐标序列传入plot()，同时保留中点标记：

from sympy import plot, sin, cos, pi, Point2D, Segment2D

x, y, angle, length = 1, 1, pi/6, 0.5
dx = length * cos(angle)
dy = length * sin(angle)
p1 = Point2D(x-dx, y-dy)
p2 = Point2D(x+dx, y+dy)
segment = Segment2D(p1, p2)

# 提取线段端点的坐标列表
x_coords = [p1.x, p2.x]
y_coords = [p1.y, p2.y]

# 创建绘图对象，先不显示
p = plot((x_coords, y_coords), line_color='blue', show=False)
# 添加中点标记
p.markers = [{'args': (x, y, 'ro')}]
# 显示图形
p.show()

方法二：直接使用Matplotlib绘制 #

由于SymPy的plot底层依赖Matplotlib，直接用Matplotlib可以更灵活处理几何对象：

from sympy import sin, cos, pi, Point2D, Segment2D
import matplotlib.pyplot as plt

x, y, angle, length = 1, 1, pi/6, 0.5
dx = length * cos(angle)
dy = length * sin(angle)
p1 = Point2D(x-dx, y-dy)
p2 = Point2D(x+dx, y+dy)
segment = Segment2D(p1, p2)

# 将SymPy点的精确值转为浮点数，适配Matplotlib
x1, y1 = float(p1.x), float(p1.y)
x2, y2 = float(p2.x), float(p2.y)

# 绘制线段
plt.plot([x1, x2], [y1, y2], 'b-', label='Segment')
# 绘制中点
plt.plot(x, y, 'ro', label='Midpoint')
# 添加图例和网格
plt.legend()
plt.grid(True)
# 显示图形
plt.show()

额外说明 #

如果需要保留SymPy的精确计算特性，也可以先将Segment2D转换为参数化的符号表达式，再传入plot()：

from sympy import plot, sin, cos, pi, Point2D, Segment2D, symbols

t = symbols('t')
# 参数化线段：t从0到1，对应从p1到p2的点
param_segment = p1 + t*(p2 - p1)
# 提取x和y的参数表达式
x_expr = param_segment.x
y_expr = param_segment.y

# 绘制参数曲线，t范围0到1
p = plot((x_expr, y_expr, (t, 0, 1)), line_color='blue', show=False)
p.markers = [{'args': (x, y, 'ro')}]
p.show()

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