How to plot square function with matplotlib

Обновить

November 2018

Просмотры

2.9k раз

2

У меня есть список значений, которые чередуются между 0 и 1, например, [0,1,0,1,0] и я хочу, чтобы построить график их так, что они выглядят как квадратные волны с использованием Matplotlib для питона. У меня это до сих пор:

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()

Это дает мне выход , как это этот:

Как сделать так, вместо того, чтобы идти на угол между точками линия остается плоским?

Я нашел этот пост , но он имеет дело больше с анимацией и не только построением функции.

1 ответы

9

Соответствующий бит с этого поста вы ссылаетесь является drawstyle:

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