Python Sympy：优雅解决动态直线交点的实时追踪

最近在做一个数学的动画时，遇到一个需求。
简单来说，就是一个直线l1l_1l1​在移动时，它与另一个直线l2l_2l2​的交点如何定位？

制作动画的时候，当直线l1l_1l1​移动时，需要不断重新计算它与l2l_2l2​的新交点，然后在动画中渲染出来。

1. 问题描述

任意点AAA和BBB确定一条直线l1l_1l1​，点CCC和DDD确定一条直线l2l_2l2​。
两直线的交点OOO。

当点BBB移动时，动态绘制交点OOO的变化。

2. 解决方法

解决的思路不难，首先计算交点OOO的坐标，
1-1. 根据点A(x1,y1)A(x_1, y_1)A(x1​,y1​)和点B(x2,y2)B(x_2, y_2)B(x2​,y2​)可以计算出直线l1l_1l1​的方程（y=k1x+b1y = k_1x+b_1y=k1​x+b1​）
1-2. 根据点C(x3,y3)C(x_3, y_3)C(x3​,y3​)和点D(x4,y4)D(x_4, y_4)D(x4​,y4​)可以计算出直线l2l_2l2​的方程（y=k2x+b2y = k_2x+b_2y=k2​x+b2​）
1-3. 根据l1l_1l1​的方程和l2l_2l2​的方程可以计算出点OOO的坐标

当点BBB不断变化时：
2-1. 根据点AAA和不断变化的点BBB重新计算l1l_1l1​的方程（y=k1x+b1y = k_1x+b_1y=k1​x+b1​）
2-2. 根据l1l_1l1​和l2l_2l2​的方程不断重新计算点OOO的坐标

这里麻烦的地方在于需要推导出下面2个公式，然后用代码来实现：

1. 根据两个点的坐标推导出直线方程中的斜率kkk和截距bbb
2. 根据两条直线的方程推导出交点坐标

如果用Python Sympy库的话，那么一下就简单很多了。

2.1. 两点坐标求直线

根据两点坐标推导出直线斜率kkk和截距bbb：

from sympy import Symbol, solve

def get_line(p1, p2):
    k = Symbol("k")
    b = Symbol("b")

    # 代入点p1坐标
    expr1 = p1[0] * k + b - p1[1]
    # 代入点p2坐标
    expr2 = p2[0] * k + b - p2[1]

    ret = solve((expr1, expr2), dict=True)
    return {"k": ret[0][k], "b": ret[0][b]}

2.2. 两条直线求交点

根据两条直线的斜率kkk和截距bbb推导交点的坐标：

def cross_point(l1, l2):
    x = Symbol("x")
    y = Symbol("y")

    # 直线l1的方程
    expr1 = l1["k"] * x + l1["b"] - y
    # 直线l2的方程
    expr2 = l2["k"] * x + l2["b"] - y
    ret = solve((expr1, expr2), dict=True)

    return np.array((float(ret[0][x]), float(ret[0][y]), 0))

2.3. 实现效果

利用上面两个函数，可以很容易得到两条直线交点的坐标，渲染后的效果如下：

3. 总结

利用Sympy带来的最大便利在于不需要推导公式，只要列出方程，交由Sympy去推导结果即可。
这样，我们在编写一些几何方面的运算时，并不需要了解多少数学的知识，也不用去记住那些求根公式等等。

同时，也可以大大简化代码，比如上面示例中封装的两个函数（get_line 和 cross_point），
可以看出，不仅实现的代码很简短，可读性也提高很多，几乎不需要多少数学知识就能看懂。

---------------------------END---------------------------

题外话

感兴趣的小伙伴，赠送全套Python学习资料，包含面试题、简历资料等具体看下方。

👉CSDN大礼包🎁：全网最全《Python学习资料》免费赠送🆓！（安全链接，放心点击）

一、Python所有方向的学习路线

Python所有方向的技术点做的整理，形成各个领域的知识点汇总，它的用处就在于，你可以按照下面的知识点去找对应的学习资源，保证自己学得较为全面。

二、Python兼职渠道推荐*

学的同时助你创收，每天花1-2小时兼职，轻松稿定生活费.

三、最新Python学习笔记

当我学到一定基础，有自己的理解能力的时候，会去阅读一些前辈整理的书籍或者手写的笔记资料，这些笔记详细记载了他们对一些技术点的理解，这些理解是比较独到，可以学到不一样的思路。

四、实战案例

纸上得来终觉浅，要学会跟着视频一起敲，要动手实操，才能将自己的所学运用到实际当中去，这时候可以搞点实战案例来学习。

👉CSDN大礼包🎁：全网最全《Python学习资料》免费赠送🆓！（安全链接，放心点击）

若有侵权，请联系删除