如何用matplotlib绘制平方函数

Question

我有一个在0和1之间交替的值列表,例如[0,1,0,1,0],我想绘制它们,因此它们使用matplotlib for python显示为方波.到目前为止我有这个:

input_amp = [1,0,1,0,1,0,1,0,1,0]
plt.plot(input_amp, marker='d', color='blue')
plt.title("Waveform")
plt.ylabel('Amplitude')
plt.xlabel("Time")
plt.savefig("waveform.png")
plt.show()

这给了我这样的输出 :

我怎么做到这一点而不是在线保持平坦的点之间的角度？

我找到了这篇文章,但它更多地涉及动画而不仅仅是绘制函数.

Answer 1

您引用的那篇文章的相关位是drawstyle:

 plt.plot(input_amp, marker='d', color='blue', drawstyle='steps-pre')