Sottorange di un array (di numpy) in base a condizione

Esempio: funzione quadratica, selezione dell'intervallo fra 0 e 5

import matplotlib.pyplot as plt
import numpy as np

# grafico x e y
x=np.linspace(-5,10,100)
y=x**2
plt.plot(x,y)

## selezione
x2 = x[(x>0) & (x<5)]
y2 = y[(x>0) & (x<5)]

plt.plot(x2,y2,linewidth=5,color='red')
plt.show()