Update beamtheta.py

beamtheta.py

@@ -46,14 +46,16 @@ for angle in np.arange(0,360,1):
 plt.plot(np.arange(0,2*framesize,1),norm(z[framesize,:]), color='blue', linestyle='--', linewidth=3, marker="o", markerfacecolor='red')
 plt.ylim(bottom=0)
 
-plt.xlabel('${x}$',size=11)
-plt.ylabel('${y}$',size=11)
+plt.xlabel('${x[mm]}$',size=11)
+plt.ylabel('${y[mm]}$',size=11)
+plt.title('Sliced beam distribution at x=23mm')
 
+
+plt.show()
+plt.imshow(z)
+plt.xlabel('${x[mm]}$',size=11)
+plt.ylabel('${y[mm]}$',size=11)
+plt.title('2D Gaussian beam distribution z(x,y)')
+plt.clim(0,100)
+plt.colorbar()
 plt.show()
-#plt.imshow(z)
-#plt.xlabel('${x}$',size=11)
-#plt.ylabel('${y}$',size=11)
-#plt.title('2D Gaussian beam distribution z(x,y)')
-#plt.clim(0,100)
-#plt.colorbar()
-#plt.show()