I am trying to rotate an image about a specific line drawn within the image. Think of the image of a random 2D numpy array. I have two specific points in the array through which I would like to plot a line. I want to rotate the array/image in such a way that this line forms the new x-axis. How do I do it?
Some code explanation:
import numpy as npimport matplotlib.pyplot as pltfrom scipy import statsx = np.linspace(-5,5,8)y = np.linspace(-25,25,7)z = np.random.randn(7,8)*0.5z = z/np.max(z)z[5,6] = 2; z[1,1] = 2plt.pcolormesh(x,y,z); plt.colorbar()
Now I want to draw a line between the the points with maximum z-values and rotate the array such that this line is the new x-axis.
I try this by first getting the end points of the line. These are usually centroids of large z-values and then get the slope of the line of best fit through these points.
# get endpoints which lie at the very end of the gaussian distributionline_endpoints = np.where(z>z.mean()+3*z.std(), 1, 0)yline, xline = np.nonzero(line_points)fitline = stats.linregress(x=xline, y=yline)Now, fitline gives me the slope and intercept of the best-fit line. How can I rotate the original array z using this slope such that the selected end-points form the new x-axis? I want to look at such an image. I checked out scipy docs but couldn't find anything suitable.