Python无额外库计算任意朝向矩形最长边中心线的问题咨询

现有代码存在的问题 #

• 顶点索引硬编码，仅支持当前示例的轴对齐矩形，旋转后顶点顺序变化会直接失效
• 存在大量重复的坐标判断逻辑，维护成本高
• 长度比例阈值abs(1-l1/l3) < 0.3通用性差，仅适配接近正方形的矩形场景

通用实现思路 #

不依赖矩形朝向的核心逻辑：矩形任意对边的中点连线即为对应边方向的中心线，只要找到最长边的对边，计算两组中点即可得到目标中心线，具体步骤：

1. 将4个顶点按输入顺序配对存储，遍历计算四条边的长度
2. 定位最长边的索引，根据矩形顶点顺/逆时针排列的特性，最长边的对边索引为 (最长边索引 + 2) % 4
3. 分别计算最长边的中点、对边的中点，两点连线即为所求中心线

优化后代码（无第三方依赖，仅matplotlib用于可视化） #

import matplotlib.pyplot as plt

# 可替换为任意朝向矩形的顶点坐标（顺/逆时针排列即可）
x = [4, 4, -4, -4]
y = [8, -5, -5, 8]
# 测试旋转后的矩形可替换为下方坐标
# x = [0, 3, 1, -2]
# y = [0, 4, 6, 2]

points = list(zip(x, y))

# 计算两点距离的工具函数
def get_dist(p1, p2):
    return ((p1[0]-p2[0])**2 + (p1[1]-p2[1])**2)**0.5

# 计算四条边的长度
edge_lens = []
for i in range(4):
    p1 = points[i]
    p2 = points[(i+1)%4]
    edge_lens.append(get_dist(p1, p2))

# 找到最长边的索引
max_edge_idx = edge_lens.index(max(edge_lens))
# 最长边的对边索引
opposite_edge_idx = (max_edge_idx + 2) % 4

# 计算最长边的中点
p_a1 = points[max_edge_idx]
p_a2 = points[(max_edge_idx+1)%4]
mid_a = ((p_a1[0] + p_a2[0])/2, (p_a1[1] + p_a2[1])/2)

# 计算对边的中点
p_b1 = points[opposite_edge_idx]
p_b2 = points[(opposite_edge_idx+1)%4]
mid_b = ((p_b1[0] + p_b2[0])/2, (p_b1[1] + p_b2[1])/2)

# 可视化
# 闭合矩形边界
plt.plot(x + [x[0]], y + [y[0]])
# 绘制中心线
plt.plot([mid_a[0], mid_b[0]], [mid_a[1], mid_b[1]])
plt.show()

优化点说明 #

• 完全适配任意朝向的矩形，仅要求输入顶点为顺时针/逆时针的有序排列，无额外限制
• 无冗余重复代码，逻辑可扩展性强，如需适配其他四边形也可快速调整
• 取消了不可靠的阈值判断，所有计算基于矩形几何特性，准确率100%

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