我正在使用 sympy 从多个表达式中收集术语,并希望在表格中格式化结果(在 ipython-notebook 中),术语在最左列下方,每个后续列代表一个表达式。该列中的条目来自dict
返回的sympy.collect(syms, evaluate=False)
到目前为止,我有:
from IPython.display import display, Latex
import pandas as pd
import sympy as sym
sym.init_printing()
x,y,z = sym.symbols('x,y,z')
da,db,dc = sym.symbols('{\Delta}a {\Delta}b {\Delta}c ' )
e_list = []
d_list = []
e_list.append(da*2*x + da*(y - 2) + db*3*z + dc*(x+y))
e_list.append(dc*z + dc*x + da*x + db*(z+2))
for e in e_list:
display(e.expand().collect((x,y,z)))
d_list.append(e.expand().collect((x,y,z),evaluate=False))
df = pd.DataFrame(d_list).T
数据框显示为我想要的,除了条目是在原始乳胶中。
我虽然以下方法可行:
Latex(df.to_latex())
但我得到的只是被一个盒子包围的乳胶代码。
编辑:这似乎是 ipython 和乳胶表的一个已知问题,请参见此处:
我有两种不同的解决方法。使用 unicode:
sym.init_printing(use_latex=False)
...
da,db,dc = sym.symbols('∆a ∆b ∆c' )
或者将表格显示为 sympy 矩阵:
terms = [x,y,z]
d_list = [(e.expand().collect((terms),evaluate=False)) for e in e_list]
mterms = sym.zeros(M.shape[0],len(terms)+1)
key1 = d_list[0].keys()[0]
terms.insert(0,key1)
for i in range(mterms.shape[0]):
for j in range(mterms.shape[1]):
try:
mterms[i,j] = d_list[i][terms[j]]
except:
mterms[i,j] = 0
mterms