Taking elements in solve and multiplying to a polynomial

asked 2021-03-18 18:56:33 +0100

whatupmatt
175 ●7 ●8 ●19

updated 2021-03-18 22:00:08 +0100

slelievre
17704 ●22 ●160 ●350 http://carva.org/samue...

So I want a function that solves a system of two equations and the move those numerical results to polynomials f and g.

def PolynomialCoefficient(fcn1, fcn2):
    var ('a b')
    eqn1 = 2*a + 2*b == 0
    eqn2 = 3*a - b == 1
    List = solve([eqn1, eqn2], a, b)

Now I want to move this to a*f(x)+b*g(x) without hitting solve each time and manually typing what a and b are. For example, PolynomialCoefficient(x, y) should output 1/4 x - 1/4 y.

The issue is len(List) outputs 1. If it was length 2 with a in first slot and b in 2nd slot, I could do

    Poly = List[0]*fcn1 + List[1]*fcn2
    return Poly

Any ideas on how to move a and b?

edit retag flag offensive close merge delete

add a comment

2 Answers

Sort by » oldest newest most voted

answered 2021-03-18 22:02:59 +0100

slelievre
17704 ●22 ●160 ●350 http://carva.org/samue...

updated 2021-03-18 22:03:12 +0100

The solve function has an optional argument solution_dict.

Using it, solutions are returned as dictionaries, which makes it easier to access the solved values of each unknown.

Here is a way to use that in the function from the question.

def polynomial_coefficient(fcn1, fcn2):
    a, b = SR.var('a, b')
    eqn1 = 2*a + 2*b == 0
    eqn2 = 3*a - b == 1
    sol = solve([eqn1, eqn2], a, b, solution_dict=True)[0]
    return sol[a]*fcn1 + sol[b]*fcn2

edit flag offensive delete link

Comments

Great thanks

whatupmatt ( 2021-03-20 05:19:01 +0100 )edit

add a comment

answered 2021-03-18 21:39:21 +0100

Max Alekseyev

6488 ●9 ●47 ●153

updated 2021-03-18 21:39:40 +0100

Your List is a list of solutions with respect to a,b. If you want just one solution you can re-define it as

List=solve([eqn1,eqn2],a,b)[0]

and then proceed as in your code; or if you want to get Poly as a list of polynomials (one for each solution), you can define it as

Poly=[ List[0]*fcn1+List[1]*fcn2 for List in solve([eqn1,eqn2],a,b) ]

edit flag offensive delete link

Comments

Thanks for your help.

whatupmatt ( 2021-03-20 05:19:10 +0100 )edit

add a comment

Your Answer

Please start posting anonymously - your entry will be published after you log in or create a new account.

Add Answer

Taking elements in solve and multiplying to a polynomial

2 Answers

Comments

Comments

Your Answer

Question Tools

Stats

Related questions

Taking elements in solve and multiplying to a polynomial edit

2 Answers

Comments

Comments

Your Answer

Question Tools

Stats

Related questions

Taking elements in solve and multiplying to a polynomial