Python - Graficar Funciones a trozos

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(-2, 6)

y = np.piecewise(x, [ (x <= 0) & (x >= -2), (x <= 1) & (x >= 0),(x <= 3) & (x >= 1), (x <= 5) & (x >= 3)], [-2, lambda x: x+2, lambda x: 0.5*x-1.5,

lambda x: -0.5*x+2.5])

plt.plot(x, y)

plt.grid()

plt.show()

import numpy as np

import matplotlib.pyplot as plt

x1 = np.linspace(-2, 0, num=3)

x2 = np.linspace(0, 1, num=2)

x3 = np.linspace(1, 3, num=3)

x4 = np.linspace(3, 5, num=3)

plt.plot(x1, [2 for x in x1], 'g',  label='y = 2, x en [-2, 0]')

plt.plot(x2, [x + 2 for x in x2], 'b', label='y = x + 2, x en [0, 1]')

plt.plot(x3, [.5 * x - 1.5 for x in x3], 'r', label='y = .5*x - 1.5, x en (1, 3]')

plt.plot(1, -1, 'ro', markersize=5, fillstyle='none', markerfacecolor='w')

plt.plot(x4, [.5 * x + 2.5 for x in x4], 'm', label='y = .5*x +2.5, x en (3, 5]')

plt.plot(3, 4, 'mo', markersize=5, fillstyle='none', markerfacecolor='w')

plt.legend(bbox_to_anchor=(0., 1.02, 1., .102), loc='lower left',

           ncol=2, mode="expand", borderaxespad=0.)

plt.grid()

plt.show()