Python | 含多个未知参数方程的曲线绘制

1 背景

前几天，一位高中同（ji）学（you）给我发来一个问题，具体见下面图片：

• 上述除了Vo和fu，其余字母均已知，而且fu并不是u的函数，就是一个符号。
• 现在的问题是希望得到Vo和fu的关系：
  • 最简单的方法当然是直接解出Vo=XXXfu，即类似于y=f(x)，但是上述方程过于复杂，无法直接求解得到，或者说很难得到！
  • 那具体该如何处理呢？退而求其次，我不需要求出两者具体的一个关系，而是直接画出图像来，通过图形直观得出两者关系！
  • 上述简化的思路我同学也是这么想的，现在的关键问题就是：如何画出两者的图形来呢？
• 绘制两个变量的关系图，最简单的方式就是类似于y=f(x)，然后对x取一个范围的值，y也就有了，然后直接画散点图。但没有这个形式是否可以绘制呢？当然是可以的！只不过需要花一点功夫！下面具体解析

2 代码解决

将上述的问题进行简化，简化后的问题为：

• 绘制f = 2v + f(v-1) 的图像
• 即f=2v/(2-v)

代码思路：

• 首先通过描点法得到两个变量的一一对应关系
• 首先给定一个变量值，然后通过solve函数反解出另一个变量值
• solve函数两个参数：第一个参数方程，另其为0。第二个参数是需要求的变量值
• 代码加入异常体

f_list = []
v_list = []
from sympy import *
for f in range(-100,100):
    f_list.append(f)
    # 设v为自变量
    v = Symbol('v')
    f = v * 2 + f * (v-1)
    # solve第一个参数是方程，另其为0，求解solve的第二个参数
    res = solve(f - v * 2 - f * (v-1) ,v)
#     print(res)
    try:
        v_list.append(res[1]) # 其中res结果为列表，第一个值都是1，取第二个值
    except Exception as e:
        print(e)
        v_list.append(0)

# 绘图
import matplotlib.pyplot as plt
plt.plot(v_list, f_list)
plt.show()

list index out of range
list index out of range

3 参考

• http://www.cocoachina.com/articles/89264