温馨提示×

怎么使用SciPy解决非线性方程组

小亿
88
2024-05-24 16:30:14
栏目: 编程语言

要使用SciPy解决非线性方程组,可以使用scipy.optimize模块中的root函数。以下是一个简单的示例代码,演示如何使用SciPy解决非线性方程组:

import numpy as np
from scipy.optimize import root

# 定义一个包含两个未知数的非线性方程组
def func(x):
    return [x[0] + 2*x[1] - 2, x[0]**2 + 4*x[1]**2 - 4]

# 初始猜测值
x0 = np.array([0, 0])

# 使用root函数求解非线性方程组
sol = root(func, x0)

# 输出结果
print(sol.x)

在这个示例中,func函数定义了一个包含两个未知数的非线性方程组。我们使用root函数来求解该方程组,并提供一个初始猜测值x0。最后,打印出求解得到的结果sol.x,即方程组的解。

0