如何绘制具有不同 sin 函数值的方程？

Question

我正在尝试以类似于链接中的图中所示的方式绘制函数：

此图所基于的等式是：y= sin^2θ x /(e^x +1)

我之前使用以下代码绘制了上面没有正弦函数的方程：

# Generate some data.
x = np.logspace(-1, 3, 100)
y= x * 1/ (np.exp(x) + 1)


plt.loglog(x, y)
plt.autoscale(enable=True, axis='x', tight=True) # optionally set a tight x-axis
#adding labels to axis
plt.xlabel('y=E/T')
plt.ylabel('$f_s$')
plt.show()

但是我怎样才能用不同的 sin 函数来绘制这个函数。我已尝试应用以下链接中使用的内容，但我不明白代码：How can I convert numbers to a color scale in matplotlib?

我是 python 新手，代码中几乎没有任何描述。

---

我也尝试应用How do I plot a function with different values of a parameter on the same plot in gnuplot 4.4? 中显示的方法，但我不断收到相同的错误消息“无效的 synthax pyplate E”

Answer 1

theta = [...] # your values of theta you'd like to plot
x = np.linspace(-1, 3, 100) # Or np.logspace, as you want. You'll have to change the parameters though, if you want to display properly
y = x / (np.exp(x)+1)

for t in theta :
  plt.plot(x, np.sin(2*t)*y, label = 'theta = %s' % t)
plt.legend()
plt.show()

theta = [1,2,3] 的结果：

如果你想显示一个颜色图，你可以这样做（代码取自here）：

import matplotlib
norm = matplotlib.colors.Normalize(vmin=np.min(theta), vmax=np.max(theta))
c_m = matplotlib.cm.cool
s_m = matplotlib.cm.ScalarMappable(cmap=c_m, norm=norm)
s_m.set_array([])

for t in theta :
  plt.plot(x, np.sin(2*t)*y, color=s_m.to_rgba(t))

plt.colorbar(s_m)
plt.show()

结果：