python计算模型psi