绘制图形非线性函数

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

我不想告诉你这个消息，但从技术上讲，阶数大于1的多项式也是非线性的。

当您在matplotlib中绘图时，您实际上是在以足够美观的分辨率提供谨慎的x和y值。在本例中，您选择了xp来确定为抛物线绘制的点。然后调用p(xp)在这些位置生成一个y值数组。

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

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