2D colormap of a polar coordinates function

I have a function defined as:

which I can visualise in 1D using:

r = -2:0.001:2;
a1 = 5;
I = a1 * exp(-2*(r).^2);
plot(r,I);

However, this function is radially symmetric, so I should be able to plot it in 2D and make it look like this:

How could I do this? I would like it to remain a plot of r (on the horizontal axes) against I(r).

Answer 1

[Ameer Hamza]

You will need to use

x = -2:0.02:2;
[x,y] = meshgrid(x);
a1 = 5;
r2 = x.^2+y.^2;
I = a1 * exp(-5*r2);
surf(x,y,I);
shading interp
colormap(jet)

[Matthew Barber]

Perfect, thank you!

[KALYAN ACHARJYA]

Thanks, you answered it exactly sir

[Ameer Hamza]

I am glad to be of help! :)