遍历折线图

Question

我现在正在尝试演示一个成本函数，想知道是否有一种方法可以通过在相同的子图和图中添加图来遍历抛物线。

def costfuntion(b, a):
    # b Value
    x = np.linspace(b*(-b), b*(b), 100)
    y = (x - a)**2
    return y

我的剧情尝试：

plt.plot(costfuntion(20,5))
plt.ylabel("Cost Value");

#b = 'some b value'
plt.plot(b, marker='o', color='b')

What I'm trying to mimic (大约: 1:13)

​

​

Answer 1

由于y的值取决于a的值，因此您需要为y查找指定一个a。请考虑以下几点：

def costfunction(b, a):
    # b Value
    x = np.linspace(b*(-b), b*(b), 100)
    y = (x - a)**2
    return x, y
a = 5
c = costfunction(20, a)
plt.plot(c[0], c[1], linestyle='-', linewidth=1)
plt.ylabel("Cost Value");

b = 100
yb = (b - a)**2    # Find the corresponding y-value
plt.plot(b, yb, marker='o', color='b')
plt.show()

这会给你

​

​

您可能还注意到，我修改了costfunction定义以返回matplotlib值，否则x将只使用它喜欢的任何值。

Answer 2

def costfuntion(b, a):
    # b Value
    x = np.linspace(b*(-b), b*(b), 100)
    y = (x - a)**2
    return x, y

x, y = costfuntion(20,5)
plt.plot(x, y)
for i in range(0, len(x), 2):
    plt.plot(x[i], y[i], marker='o', color='b')

更改成本函数以同时返回要绘制的函数的x和y，并使用此信息在函数上绘制点。