numpy - 不能将 set_ydata 用于 matlibplot 中的“茎”图

Question

我无法正确使用 set_ydata 方法。我有以下代码（我使用 IPython 笔记本）

%pylab wx
x = arange(20.)
y_old = x**2
y_new = 1.3*y_old

markerline, stemlines, baseline = stem(x,y_old)
markerline.set_ydata(y_new)

结果如下图：

我试图添加：

stemlines.set_ydata(y_new)
baseline.set_ydata(y_new)

但程序崩溃了。可以做些什么来刷新主干图的所有元素？

Answer 1

您需要循环茎线，并为每一行设置 y_data：

x = arange(20.)
y_old = x**2
y_new = 1.3*y_old

markerline, stemlines, baseline = stem(x,y_old)
markerline.set_ydata(y_new)
for line, y in zip(stemlines, y_new):
    line.set_ydata([0, y])