如何用Python在高维两点间拟合直线并随机采样中间点

原理说明 #

两点之间的线段可以用参数化方程表示：对于点p0和p1，线段上任意一点p(t)满足：

p(t) = p0 + t*(p1 - p0)
其中t是0到1之间的任意实数，当t=0时得到p0，t=1时得到p1，取0到1之间的随机t就能得到线段上的随机点。

实现代码 #

方法一：用numpy（推荐，适合高维向量运算）

import numpy as np

# 定义两个5维点（将原格式的空格替换为逗号）
p0 = np.array([2, 4, 1, 3, 9])
p1 = np.array([5, 2, 1, 0, 8])

# 生成0到1之间的随机参数t
t = np.random.uniform(0, 1)

# 计算随机采样点p2
p2 = p0 + t * (p1 - p0)

print("采样得到的点p2：", p2)

方法二：纯Python列表实现（无需第三方库）

import random

p0 = [2, 4, 1, 3, 9]
p1 = [5, 2, 1, 0, 8]

t = random.uniform(0, 1)
# 逐个维度计算采样点
p2 = [p0[i] + t * (p1[i] - p0[i]) for i in range(len(p0))]

print("采样得到的点p2：", p2)

补充说明 #

• 两种方法都能生成线段上的随机点，numpy版本在处理更高维数据时效率更优
• 每次运行代码会得到不同的p2，因为t是随机生成的
• 若需要批量生成多个采样点，只需循环执行采样逻辑即可

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