如何在Python中将复数矩阵转为LaTeX可用的科学计数法格式

问题描述 #

6×6复数矩阵A_q1a的特征向量矩阵eigen_vectors在LaTeX中显示过于冗长，需要转换为保留3位有效数字的科学计数法格式，且要求虚部为0时必须显示为0i。尝试过多种转换方法，要么虚部被自动移除，要么格式完全没有变化。

原始代码如下：

import scipy.io
import numpy as np
from print_latex import print_latex as pl
from latexifier import latexify

mat = scipy.io.loadmat('system_q1a.mat')
np.array(mat)
A_q1a=mat["A_q1a"]
[eigen_values,eigen_vectors]=np.linalg.eig(A_q1a)
imag_eigen_values=np.imag(eigen_values)
real_eigen_values=np.real(eigen_values)
f_imag_eigen_values=imag_eigen_values/(2*np.pi)
damping_eigen_values=-real_eigen_values/(np.sqrt(np.power(real_eigen_values,2) + np.power(imag_eigen_values,2)))

#Convert to Latex
#pl(eigen_values,'bmatrix')
pl(eigen_vectors,'bmatrix')
#pl(imag_eigen_values,'bmatrix')
#pl(real_eigen_values,'bmatrix')
#pl(f_imag_eigen_values,'bmatrix')
#pl(damping_eigen_values,'bmatrix')

def print_latex(matrix,arraytype):
    print(latexify(matrix,arraytype=arraytype,fmt="%.1e"))
    return

解决方法 #

核心是自定义复数元素的格式化逻辑，替代原有工具的默认处理规则，确保满足格式要求：

修改后的代码 #

import scipy.io
import numpy as np

def format_complex_num(z):
    # 按3位有效数字、科学计数法格式化实部和虚部
    real_str = np.format_float_scientific(np.real(z), precision=3, exp_digits=1)
    imag_val = np.imag(z)
    # 处理虚部为0的情况，强制显示0i
    if np.isclose(imag_val, 0):
        imag_str = "0"
    else:
        imag_str = np.format_float_scientific(imag_val, precision=3, exp_digits=1)
    
    # 组合LaTeX格式的复数表达式
    if np.isclose(np.real(z), 0):
        return f"{imag_str}i"
    elif np.isclose(imag_val, 0):
        return f"{real_str} + {imag_str}i"
    else:
        return f"{real_str} + {imag_str}i"

def matrix_to_latex(matrix, arraytype='bmatrix'):
    # 逐行生成LaTeX矩阵内容
    row_lines = []
    for row in matrix:
        formatted_elements = [format_complex_num(elem) for elem in row]
        row_lines.append(" & ".join(formatted_elements))
    # 拼接成完整的LaTeX矩阵环境
    return f"\\begin{{{arraytype}}}\n" + "\\\\\n".join(row_lines) + f"\n\\end{{{arraytype}}}"

# 加载数据并计算特征向量
mat = scipy.io.loadmat('system_q1a.mat')
A_q1a = mat["A_q1a"]
_, eigen_vectors = np.linalg.eig(A_q1a)

# 生成并输出符合要求的LaTeX代码
print(matrix_to_latex(eigen_vectors))

关键说明 #

1. format_complex_num函数：
   • 使用np.format_float_scientific严格控制3位有效数字的科学计数法格式
   • 通过np.isclose判断虚部是否为0，强制输出0i
   • 根据实部和虚部的取值情况，生成简洁且符合要求的复数表达式
2. matrix_to_latex函数：
   • 遍历矩阵的每一行，将每个元素转换为格式化后的字符串
   • 拼接成LaTeX矩阵环境（默认bmatrix），直接输出可用于LaTeX文档的代码

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