python - 極座標でのMatplotlib密度プロット?

Question

極座標の分布の値に対応するエントリを持つ txt ファイルとして保存された配列があります。したがって、次のようになります。

  f(r1,theta1) f(r1, theta2) ..... f(r1, theta_max)
  f(r2,theta1) f(r2, theta2) .....        .
        .                                 .
        .                                 .
        .                                 .
  f(r_max,theta1) .................f(r_max, theta_max)

f の密度プロットを実行したい (f が高いほど、色を赤くしたい)。matplotlib でこれを行う方法はありますか? 私はこれにほとんど慣れていないので、明示的なコードは役に立ちます。

Answer 1

この例でaは、あなたは theta1...thetan ですか、bあなたの r1...rncはあなたの f(a, b) です:

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm as cm

#fake data:
a = np.linspace(0,2*np.pi,50)
b = np.linspace(0,1,50)
A, B = np.meshgrid(a, b)
c = np.random.random(A.shape)

#actual plotting
ax = plt.subplot(111, polar=True)
ax.set_yticklabels([])
ctf = ax.contourf(a, b, c, cmap=cm.jet)
plt.colorbar(ctf)

基本的に、極軸の塗りつぶされた等高線図。で代替カラーマップを指定できますcmap=...。cm.jetは青から赤に変化し、赤が最大値です。