Circular heatmap python

from pylab import *
import numpy as np
from scipy.interpolate import griddata

#create 5000 Random points distributed within the circle radius 100
max_r = 100
max_theta = 2.0 * np.pi
number_points = 5000
points = np.random.rand(number_points,2)*[max_r,max_theta]

#Some function to generate values for these points, 
#this could be values = np.random.rand(number_points)
values = points[:,0] * np.sin(points[:,1])* np.cos(points[:,1])

#now we create a grid of values, interpolated from our random sample above
theta = np.linspace(0.0, max_theta, 100)
r = np.linspace(0, max_r, 200)
grid_r, grid_theta = np.meshgrid(r, theta)
data = griddata(points, values, (grid_r, grid_theta), method='cubic',fill_value=0)

#Create a polar projection
ax1 = plt.subplot(projection="polar")
ax1.pcolormesh(theta,r,data.T)
plt.show()

Source: stackoverflow.com

Similar codes

0 vote215 views1 answer
Convert torch.nn.Embedding layer to numpy array
Python
1 vote336 views1 answer
Uninstall poetry
Python
4 votes262 views1 answer
Django pandas queryset
Python
11 votes201 views3 answers
How to print all elements of a dictionary in python
Python
1 vote229 views1 answer
Pandas dataframe remove rows by column value
Python
28 votes183 views1 answer
'utf-8' codec can't decode byte 0x85 in position 715: invalid start byte
Python
15 votes154 views2 answers
Numpy create zero array
Python
3 votes192 views5 answers
Nltk python
Python
5 votes181 views5 answers
Closure
Python
2 votes208 views1 answer
Selenium python switch to iframe
Python