import numpy as np

import matplotlib.pyplot as plt

r $=$ np$.$linspace$(0,2,100)$

theta $=$ np$.$linspace$(0,2*$np$.$pi$,100)$

R$,$ Theta $=$ np$.$meshgrid$($r$,$ theta$)$

Psi $=($R$**2/2)*$np$.$cos$($Theta$)+$ Theta$/(2*$np$.$pi$)$

plt$.$figure$()$

plt$.$contour$($R$*$np$.$cos$($Theta$),$ R$*$np$.$sin$($Theta$),$ Psi$)$

plt$.$show$()$