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

最近在做一个数学的动画时,遇到一个需求。

简单来说,就是一个直线 l 1 l_1 l1在移动时,它与另一个直线 l 2 l_2 l2的交点如何定位?

制作动画的时候,当直线 l 1 l_1 l1移动时,需要不断重新计算它与 l 2 l_2 l2的新交点,然后在动画中渲染出来。

1. 问题描述

任意点 A A A和 B B B确定一条直线 l 1 l_1 l1,点 C C C和 D D D确定一条直线 l 2 l_2 l2。

两直线的交点 O O O。

当点 B B B移动时,动态绘制交点 O O O的变化。

2. 解决方法

解决的思路不难,首先计算交点 O O O的坐标,

1-1. 根据点 A ( x 1 , y 1 ) A(x_1, y_1) A(x1,y1)和点 B ( x 2 , y 2 ) B(x_2, y_2) B(x2,y2)可以计算出直线 l 1 l_1 l1的方程( y = k 1 x + b 1 y = k_1x+b_1 y=k1x+b1)

1-2. 根据点 C ( x 3 , y 3 ) C(x_3, y_3) C(x3,y3)和点 D ( x 4 , y 4 ) D(x_4, y_4) D(x4,y4)可以计算出直线 l 2 l_2 l2的方程( y = k 2 x + b 2 y = k_2x+b_2 y=k2x+b2)

1-3. 根据 l 1 l_1 l1的方程和 l 2 l_2 l2的方程可以计算出点 O O O的坐标

当点 B B B不断变化时:

2-1. 根据点 A A A和不断变化的点 B B B重新计算 l 1 l_1 l1的方程( y = k 1 x + b 1 y = k_1x+b_1 y=k1x+b1)

2-2. 根据 l 1 l_1 l1和 l 2 l_2 l2的方程不断重新计算点 O O O的坐标

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

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

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

2.1. 两点坐标求直线

根据两点坐标推导出直线斜率 k k k和截距 b b b:

python 复制代码
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. 两条直线求交点

根据两条直线的斜率 k k k和截距 b b b推导交点的坐标:

python 复制代码
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),

可以看出,不仅实现的代码很简短,可读性也提高很多,几乎不需要多少数学知识就能看懂。

相关推荐
BK贝壳1 小时前
亿级用户签到位图选型复盘:用 bool-hybrid-array 扛冷启动打卡,也把它和 Roaring/Redis 的边界测清楚了
数据结构·python·性能优化·numpy
2601_966949653 小时前
如何根据研究的起止时间截取分钟级行情?Python 实战与 QuantDash 数据处理指南
开发语言·python·pandas·量化交易·1024程序员节·股票数据·quantdash
在世修行7 小时前
智学文枢:页面切换与跨页数据预填
python·桌面应用·signal·1024程序员节·pyside·跨页跳转·状态预填
2601_962885728 小时前
如何用 Python 把 K 线重采样成任意周期?
开发语言·python
打工仔折腾 AI8 小时前
在 Ubuntu ARM64 部署 Q1 V3.0 MuJoCo 模型:加载、单步推理与执行器响应解析
linux·运维·人工智能·python·深度学习·ubuntu·性能优化
信誓旦旦的程序猿9 小时前
【Python量化实战宝典 #21】北向资金跟随手册:聪明钱择时信号(Python 可跑)
python·1024程序员节·股票数据api·股票数据api接口·股票api数据接口·股票量化数据api·股票量化数据接口
严谨的麻辣烫9 小时前
静态 ISP 代理请求超时怎么排查?Python 与 curl 实战
python·http·requests·1024程序员节·网络排障
KeepRobots10 小时前
11-01 温哥华不回拨:用 60 行 Python 查你的工单系统会不会差一小时(KeepRobots 工程笔记)
python
xiao77mwq10 小时前
你以为 OCR 慢?实测 81.3 s 里全是故障代价:如何把「定格态」消掉,让耗时从 6.5 s 到 13 倍差距显出原形
python
深圳市益普科技有限公司11 小时前
半导体良率管理YMS:良率与设备参数的相关性,先拉哪几组数据
人工智能·python·算法