Revision history [back]
 | 1 | initial version |
I have no idea why do we want to really do the above, and really in this way (so we lose control on the powers of $n$ while constructing an expression in $n$, then want to get back these powers...) but here is an answer. It is designed to work with the above example. Again, i would never do this. It is a better way to use or isolate the total factor $n^{-t}$.
First of all, let us initialize the E1 without a full simplification.
sage: def desc_prod_recip(x,y):
....: t0 = 1
....: t1 = y*(y-1)/(2*x)
....: t2 = (y*(y+1)*(y-1)*(3*y-2))/(24*x^2)
....: t3 = (y^2*(y-1)^2*(y+1)*(y+2))/(48*x^3)
....: t4 = y*(y-1)*(y+1)*(y+2)*(y+3)*(15*y^3-15*y^2-10*y+8)/(5760*x^4)
....: return (x^(-y))*(t0 + t1 + t2 + t3 + t4)
....: n,t = var('n t');
....: E1 = desc_prod_recip(n,t)
sage: E1 = E1.expand()
sage: for operand in E1.operands():
....: print(operand)
....:
1/384*t^8/(n^4*n^t)
1/48*t^6/(n^3*n^t)
1/96*t^7/(n^4*n^t)
1/8*t^4/(n^2*n^t)
1/48*t^5/(n^3*n^t)
-1/576*t^6/(n^4*n^t)
1/2*t^2/(n*n^t)
-1/12*t^3/(n^2*n^t)
-1/16*t^4/(n^3*n^t)
-1/30*t^5/(n^4*n^t)
1/(n^t)
-1/2*t/(n*n^t)
-1/8*t^2/(n^2*n^t)
-1/48*t^3/(n^3*n^t)
-5/1152*t^4/(n^4*n^t)
1/12*t/(n^2*n^t)
1/24*t^2/(n^3*n^t)
1/32*t^3/(n^4*n^t)
1/288*t^2/(n^4*n^t)
-1/120*t/(n^4*n^t)
Now we can put the hands on each operand. The code has the last one in its hands, and we can use for instance...
sage: operand.factor_list()
[(n, -t - 4), (t, 1), (-1/120, 1)]
Now we can parse the "factor list" above with entries of the shape (the_factor, its_power), ask if the first component is n, if yes, record its_power, and then build the product of all not considered factors to the corresponding powers. This may be easily done, but unfortunately...
sage: 1/(n^t)
1/(n^t)
sage: _.factor_list()
[(n^t, -1)]
so the implementation needs to go recursively deeper. A first approximation but not the solution would be...
sage: ops = E1.operands()
sage: n_powers = [ power for op in ops for base, power in op.factor_list() if base == n ]
sage: n_powers = list(set(n_powers))
sage: n_powers.sort()
sage: n_powers
[-t - 4, -t - 3, -t - 2, -t - 1]
sage: # so -t is missing here already, since 1/n^t is not correctly parsed...
And now we can build a dictionary with keys the powers of $n$, values their "coefficients".
sage: dic = {}
sage: for fixed_power in n_powers:
....: dic[fixed_power] = sum([ prod([ base^power
....: for base, power in op.factor_list()
....: if (base, power) != (n, fixed_power) ])
....: for op in ops
....: if (n, fixed_power) in op.factor_list()])
....:
sage: dic
{-t - 4: 1/384*t^8 + 1/96*t^7 - 1/576*t^6 - 1/30*t^5 - 5/1152*t^4 + 1/32*t^3 + 1/288*t^2 - 1/120*t,
-t - 3: 1/48*t^6 + 1/48*t^5 - 1/16*t^4 - 1/48*t^3 + 1/24*t^2,
-t - 2: 1/8*t^4 - 1/12*t^3 - 1/8*t^2 + 1/12*t,
-t - 1: 1/2*t^2 - 1/2*t}
sage:
This is only the half of the road, but we can still ask for the difference...
sage: diff = E1 - sum([ n^key*val for key, val in dic.items() ])
sage: diff.simplify_full()
1/(n^t)
sage:
As i said, i would never program such an adventure, instead change the method to attack the problem, the situation.
