Python matplotlib绘制热成像图与极坐标柱状图完整示例

周末不加班

发布时间：2026-10-13| 5人浏览过

原创

本文介绍了使用Python matplotlib库绘制热成像图和极坐标图的两种常见方法。通过具体代码示例，展示了如何生成基于数学函数的热成像图并添加colorbar，以及如何创建极坐标下的柱状图并自定义颜色与透明度。

热成像图

%matplotlib inline
from matplotlib import pyplot as plt
import numpy as np
 
def f(x, y):
  return (1 - x / 2 + x ** 5 + y ** 3) * np.exp(-x ** 2 - y ** 2)
 
n = 10
x = np.linspace(-3, 3, 4 * n)
y = np.linspace(-3, 3, 3 * n)
X, Y = np.meshgrid(x, y)
plt.imshow(f(X, Y), cmap='hot', origin='low')
plt.colorbar(shrink=.83)
 
plt.xticks(())
plt.yticks(())

极坐标图

%matplotlib inline
from matplotlib import pyplot as plt
import numpy as np
 
ax = plt.subplot(1, 1, 1, polar=True)
#ax = plt.plot( polar=True)
N = 20
theta = np.arange(0.0, 2 * np.pi, 2 * np.pi / N)
radii = 10 * np.random.rand(N)
width = np.pi / 4 * np.random.rand(N)
bars = plt.bar(theta, radii, width=width, bottom=0.0)
 
for r,bar in zip(radii, bars):
  bar.set_facecolor(plt.cm.jet(r/10.))
  bar.set_alpha(0.5)
 
ax.set_xticklabels([])
ax.set_yticklabels([])