Dec-07-2022, 02:40 PM
I would like to rotate a function that is defined in terms of the error function, see here. I am unsure how to go about it, but I would like the resulting plot to be diagonal along the line y=-bx, where b is some arbitrary positive constant. Here is the code so far; increasing
cx
and cy
stretches the graph in the y-direction and x-direction respectively.from scipy import special import numpy as np import matplotlib.pyplot as plt u0=200 r0x=25 r0y=25 rmax=5.5 alpha=2 t=2 cx=20 cy=1 y, x = np.meshgrid(np.linspace(0, 50, 100), np.linspace(0, 50, 100)) r=np.sqrt(cx*(x-r0x)**2+cy*(y-r0y)**2) z=u0*(special.erf((rmax-r)/np.sqrt(4*alpha*t))-special.erf((-rmax-r)/np.sqrt(4*alpha*t))) plt.pcolormesh(x, y, z, cmap='jet') plt.xlim(x.min(), x.max()) plt.ylim(y.min(), y.max()) plt.xlabel('x') plt.ylabel('y') plt.title(f'Temperature at {t} [arb. unit]') cbar=plt.colorbar() cbar.set_label('Temperature') plt.clim(0,200)Below is the output.