绘制图形的非线性函数

Question

我想绘制这个函数的图形：

y = 2[1-e^(-x+1)]^2-2

当我绘制线性函数时，我使用了以下代码：

import matplotlib.pyplot as plt
import numpy as np
x = np.array(...)
y = np.array(...)
z = np.polyfit(x, y, 2)
p = np.poly1d(z)
xp = np.linspace(...)
_ = plt.plot(x, y, '.', xp, p(xp), '-')
plt.ylim(0, 200)
plt.show()

当函数为非线性时，它不起作用因为很难找到每个x，y值。

如何绘制非线性函数？

Answer 1

我不想把这个消息告诉你，但一阶以上的多项式在技术上也是非线性的。

当你在matplotlib中绘图时，你实际上是在以足够视觉愉悦的分辨率提供谨慎的x和y值。在这种情况下，您已选择xp来确定为抛物线绘制的点。然后调用p(xp)在这些位置生成一个y值数组。

没有什么可以阻止你使用简单的numpy函数为感兴趣的公式生成y值：

y = 2 * (1 - np.exp(1 - xp))**2 - 2