[python] 파이썬 sympy - 방정식 풀기
파이썬과 sympy 패키지를 이용하여 방정식을 푸는 샘플 코드를 소개합니다. 풀고자 하는 방정식은 아래와 같습니다.
아래는 코드입니다.
from sympy import Symbol, solve
x = Symbol('x')
expr = x**2 + 5*x + 4
result = solve(expr, dict=True)
print(result)
아래는 실행결과입니다.
[{x: -4}, {x: -1}]
좀 더 응용해서 연립방정식을 풀어보겠습니다.
.gif)
solve() 메소드는 위 식을 아래와 같은 형태로 취급하고 해를 구하게 됩니다.
.gif)
아래는 코드입니다.
from sympy import Symbol, solve
x = Symbol('x')
y = Symbol('y')
expr1 = 2*x + 3*y - 6
expr2 = 3*x + 2*y - 12
result = solve((expr1, expr2), dict=True)
result = result[0]
print(result)
# 구해진 해를 각 식에 넣어서 수식이 성립하는지 확인
# 0이 출력되면 방정식을 만족하는 해
print(expr1.subs({x:result[x], y:result[y]}))
print(expr2.subs({x:result[x], y:result[y]}))
아래는 실행결과입니다.
{x: 24/5, y: -6/5}
0
0
끝.
댓글
댓글 쓰기