import numpy as np
from mayavi import mlab

u,v = np.mgrid[1.2:5.0:150j, 0.0:2*np.pi:150j]

x=u*np.cos(v)
y=u*np.sin(v)
z=9/u**2

mlab.mesh(x, y, z, color=(1,1,0)) 

c1=np.linspace(1.2,5.0,num=15)
c2=np.linspace(0.0,2*np.pi,num=15)
v1=np.linspace(0,2*np.pi,num=100)
u1=np.linspace(1.2,5.0,num=100)

for c in c1:
	mlab.plot3d(c*np.cos(v1),c*np.sin(v1),9/c**2*(v1+1)/(v1+1),color=(0.2,0.2,0.2), tube_radius=None, line_width=1)
	
for c in c2:
	mlab.plot3d(u1*np.cos(c),u1*np.sin(c),9/u1**2,color=(0.2,0.2,0.2), tube_radius=None, line_width=1)

#neke parametarske v-crte
#mlab.plot3d(3*np.cos(v1),3*np.sin(t1),9/9.0*(v1+1)/(v1+1),color=(0,0,0), tube_radius=None, line_width=1)
#mlab.plot3d(np.sqrt(4.5)*np.cos(v1),np.sqrt(4.5)*np.sin(v1),9/4.5*(v1+1)/(v1+1),color=(0,0,0), tube_radius=None, line_width=1)

mlab.show()