Error on creating spiral arms for galaxy formation on python

Question

i have writing this code but in my code there is error, how can i make spiral arms from this code by used profile used below i want to create surface density profile using s