sombrero.py

#!/usr/bin/python3.13
'''
https://octave.sourceforge.io/octave/function/sombrero.html
'''
import warnings
warnings.filterwarnings("ignore")
from numpy import sin, sqrt, linspace,finfo, transpose
from re import search
from sys import version as python_version
#Workaround for matplotlib bug
if search('GCC UCRT', python_version): from PyQt5 import QtCore
from matplotlib import cm, pyplot as plt
N=1000
x=linspace(-8,8,num=N+1,endpoint=True).reshape(1,N+1)
x[0][N//2]=finfo(float).eps
y=x.transpose()
rs='sqrt(x**2 + y**2)'
r=eval(rs)
plt.colorbar(plt.axes(projection = '3d').plot_surface(x,y,sin(r)/r, cmap=cm.rainbow), shrink=0.5, aspect=5)
plt.setp(plt.title('sin(' + rs + ')/' + rs), color='b', backgroundcolor='lightgray')
plt.gcf().canvas.manager.set_window_title('Sombrero plot')
plt.tight_layout()
plt.show()