“Cómo definir la función por partes I Python” Código de respuesta

Cómo definir la función por partes I Python

>>> x = np.linspace(-2.5, 2.5, 6)
>>> np.piecewise(x, [x < 0, x >= 0], [-1, 1])
array([-1., -1., -1.,  1.,  1.,  1.])