How to use print() command and make the shape of a numpy array consistent, during integration using scipy?

Question

I tried examining how spicy.integrate.ode works. The code below is simple code to do this.

def func(t, z, p):
    x = z[0]
    y = z[1]
    print(\'x :\', x)