python - 控制 SymPy 中的公式表示

Question

我创建了一个这样的公式:

>>> a, b = symbols('a, B')
>>> i = Idx('i')
>>> bi = IndexedBase('b')[i]
>>> Sum(a*(bi/b), (i, 1, 5))
  5         
 ____       
 ╲    a⋅b[i]
  ╲   ──────
  ╱     B   
 ╱         
 ‾‾‾‾       
i = 1

虽然它是正确的，但看起来不是我想要的。我想强调一下，b[i]/B 是 a 的系数，以便在阅读这个公式时更容易理解我的想法。这是我的 Jupyter Notebook 文档中的文档目的所必需的。所以我想告诉 sympy 以某种方式表示公式:

  5         
 ____       
 ╲      b[i]
  ╲   a⋅────
  ╱      B   
 ╱         
 ‾‾‾‾       
i = 1

之后我想告诉 sympy 简化公式以尽可能少地进行操作。例如，这个公式可以这样简化:

     5         
    ____       
a   ╲      
─ ⋅  ╲   b[i]
B    ╱     
    ╱         
    ‾‾‾‾       
   i = 1

如何使用 SymPy 实现此目的？

Answer 1

我无法让你的确切代码在我的旧系统(Python 2.7.6，Sympy 0.7.4.1)上工作，所以我不得不对其进行一些修改，同时仍然试图保持非常相似的结构。

看起来像factor()可以帮助你:

from sympy import *
a, b = symbols('a, B')
i, n = symbols('i n', integer=True)
bi = Function('b')(i)
f = summation(a/b*bi, (i,1,n))
init_printing()

f:

  n         
 ____       
 ╲          
  ╲   a⋅b(i)
   ╲  ──────
   ╱    B   
  ╱         
 ╱          
 ‾‾‾‾       
i = 1       

因子(f):

    n       
   ___      
   ╲        
    ╲   b(i)
a⋅  ╱       
   ╱        
   ‾‾‾      
  i = 1     
────────────
     B