如何用Pytearcat自动将Christoffel符号分配给Python函数变量？

核心思路 #

Pytearcat计算得到的Christoffel符号存储为三维数组对象chris，对应指标结构$\Gamma^k_{ij}$（$k$为上标，$i,j$为下标）。通过遍历所有指标组合，可自动将非零符号赋值到自定义函数的变量（或字典）中，替代手动编写重复代码。

实现步骤与代码示例 #

1. 基础计算（已完成部分）

先确保已用Pytearcat完成度规定义与Christoffel符号计算：

import pytearcat as pt

# 定义时空坐标（以4维为例：t, r, θ, φ）
t, r, theta, phi = pt.coords('t,r,theta,phi')
# 自定义度规张量（替换为你的线元对应的度规）
g = pt.metric('ds2 = -dt**2 + a(t)**2*(dr**2/(1-k*r**2) + r**2*dtheta**2 + r**2*sin(theta)**2*dphi**2)')
# 计算Christoffel符号
chris = pt.christoffel()

2. 在自定义函数中自动化赋值

在geodesic函数内，通过三重循环遍历所有指标，将非零Christoffel符号批量赋值：

def geodesic():
    # 方案1：用字典存储，安全可控（推荐）
    christoffel_dict = {}
    # 定义指标范围（4维时空对应0-3，对应t,r,theta,phi）
    idx_range = range(4)
    
    for k in idx_range:
        for i in idx_range:
            for j in idx_range:
                gamma_expr = chris[k, i, j]
                # 仅保留非零符号，减少冗余
                if not gamma_expr.is_zero():
                    # 生成可读性强的变量名，如gamma_0_1_1对应Γ^t_rr
                    var_name = f"gamma_{k}_{i}_{j}"
                    christoffel_dict[var_name] = gamma_expr
                    # 方案2：直接创建局部变量（慎用，可能引发作用域问题）
                    # locals()[var_name] = gamma_expr
    
    # 调用示例：获取Γ^t_rr的表达式
    print(christoffel_dict["gamma_0_1_1"])
    # 若用方案2，可直接调用变量
    # print(gamma_0_1_1)

优化说明 #

• 如果需要更直观的字母指标命名，可将idx_range替换为字母列表：idx_list = ['t', 'r', 'theta', 'phi']，变量名变为gamma_t_r_r。
• 优先使用字典存储，避免locals()带来的潜在作用域风险，同时方便后续批量处理符号。

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