如何用Python生成表示张量及其协变导数的LaTeX与UnicodeMath代码

生成LaTeX公式代码 #

LaTeX是学术场景中表示张量和协变导数的标准格式，以下Python函数可快速生成对应公式代码，支持任意数量的上下标指标：

def generate_tensor_latex(upper_indices, lower_indices, tensor_name="T"):
    upper_str = "".join(upper_indices)
    lower_str = "".join(lower_indices)
    
    if upper_str and lower_str:
        return f"${tensor_name}^{{{upper_str}}}_{{{lower_str}}}$"
    elif upper_str:
        return f"${tensor_name}^{{{upper_str}}}$"
    elif lower_str:
        return f"${tensor_name}_{{{lower_str}}}$"
    else:
        return f"${tensor_name}$"

def generate_covariant_derivative_latex(upper_indices, lower_indices, derivative_index, tensor_name="T"):
    tensor_core = generate_tensor_latex(upper_indices, lower_indices, tensor_name).strip("$")
    return f"$\\nabla_{{{derivative_index}}} {tensor_core}$"

示例使用 #

# 生成二阶混合张量T^i_j的LaTeX代码
print(generate_tensor_latex(["i"], ["j"]))  # 输出: $T^{i}_{j}$

# 生成协变导数∇_k T^i_j的LaTeX代码
print(generate_covariant_derivative_latex(["i"], ["j"], "k"))  # 输出: $\nabla_{k} T^{i}_{j}$

你可以通过扩展upper_indices和lower_indices参数，生成多指标张量的LaTeX代码，比如传入["i", "m"]即可得到$T^{im}_j$。

生成UnicodeMath公式代码 #

UnicodeMath无需依赖LaTeX编译环境，直接用Unicode字符即可显示张量和协变导数，适合普通文本编辑器、聊天工具等场景：

def generate_tensor_unicode(upper_indices, lower_indices, tensor_name="T"):
    upper_map = {
        "i": "ⁱ", "j": "ʲ", "k": "ᵏ", "l": "ˡ",
        "m": "ᵐ", "n": "ⁿ", "1": "¹", "2": "²", "3": "³"
    }
    lower_map = {
        "i": "ᵢ", "j": "ⱼ", "k": "ₖ", "l": "ₗ",
        "m": "ₘ", "n": "ₙ", "1": "₁", "2": "₂", "3": "₃"
    }
    
    upper_str = "".join([upper_map[idx] for idx in upper_indices])
    lower_str = "".join([lower_map[idx] for idx in lower_indices])
    
    return f"{tensor_name}{upper_str}{lower_str}"

def generate_covariant_derivative_unicode(upper_indices, lower_indices, derivative_index, tensor_name="T"):
    lower_map = {
        "i": "ᵢ", "j": "ⱼ", "k": "ₖ", "l": "ₗ",
        "m": "ₘ", "n": "ₙ", "1": "₁", "2": "₂", "3": "₃"
    }
    tensor_core = generate_tensor_unicode(upper_indices, lower_indices, tensor_name)
    return f"∇{lower_map[derivative_index]} {tensor_core}"

示例使用 #

# 生成二阶混合张量T^i_j的Unicode代码
print(generate_tensor_unicode(["i"], ["j"]))  # 输出: Tⁱⱼ

# 生成协变导数∇ₖ Tⁱⱼ的Unicode代码
print(generate_covariant_derivative_unicode(["i"], ["j"], "k"))  # 输出: ∇ₖ Tⁱⱼ

若需支持更多指标，只需在upper_map和lower_map中添加对应Unicode字符映射即可。

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