pylab_examples example code: step_demo.py

(Source code, png)

import numpy as np
from numpy import ma
from matplotlib.pyplot import step, legend, xlim, ylim, show

x = np.arange(1, 7, 0.4)
y0 = np.sin(x)
y = y0.copy() + 2.5

step(x, y, label='pre (default)')

y -= 0.5
step(x, y, where='mid', label='mid')

y -= 0.5
step(x, y, where='post', label='post')

y = ma.masked_where((y0 > -0.15) & (y0 < 0.15), y - 0.5)
step(x, y, label='masked (pre)')

legend()

xlim(0, 7)
ylim(-0.5, 4)

show()

Keywords: python, matplotlib, pylab, example, codex (see Search examples)