如何找到两组三维点之间的仿射变换矩阵？

import numpy as np

def find_affine_transform(P, Q):
    # P 和 Q 是形状为 (n, 3) 的数组，包含 n 个三维点
    assert P.shape == Q.shape, "点集 P 和 Q 必须具有相同的形状"
    
    # 将点转换为齐次坐标
    P_homogeneous = np.hstack([P, np.ones((P.shape[0], 1))])
    Q_homogeneous = np.hstack([Q, np.ones((Q.shape[0], 1))])
    
    # 构建方程组 A * T = B
    A = []
    B = []
    for p, q in zip(P_homogeneous, Q_homogeneous):
        A.append([p[0], p[1], p[2], 0, 0, 0, -q[0]*p[0], -q[0]*p[1], -q[0]*p[2]])
        A.append([0, 0, 0, p[0], p[1], p[2], -q[1]*p[0], -q[1]*p[1], -q[1]*p[2]])
        A.append([0, 0, 0, 0, 0, 0, p[0], p[1], p[2]])
        B.append(q[0])
        B.append(q[1])
        B.append(q[2])
    
    A = np.array(A)
    B = np.array(B).reshape(-1, 1)
    
    # 求解线性方程组
    T = np.linalg.lstsq(A, B, rcond=None)[0]
    
    # 将 T 转换为 4x4 矩阵形式
    T = np.vstack([T[:9].reshape(3, 3), T[9:]])
    T = np.hstack([T[:3, :], T[3:, :3], T[3:, 3:]])
    
    return T

# 示例数据
P = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
Q = np.array([[2, 3, 4], [5, 6, 7], [8, 9, 10]])

T = find_affine_transform(P, Q)
print("仿射变换矩阵:\n", T)