極座標の２Dカラーマッピングを作成したいです。

Question

前提・実現したいこと

極座標の２Dカラーマッピングを作成したいです。
r=-1～1
p=0~2pi
z=r*cos(p)

動径rと回転角pの２次元極座標系においてzの値をカラーマップで表示させたいです。

簡潔なコードを教えてください。

発生している問題・エラーメッセージ

エラーメッセージ

該当のソースコード

Python

試したこと

下記コードを作成しましたが、うまくいきませんでした。
修正箇所をご教示ください。お願いいたします。

import numpy as np
import matplotlib.pyplot as plt

r = np.linspace(0, 1.0, 50)
p = np.linspace(0, 2*np.pi, 50)

x=rnp.cos(p)
y=rnp.sin(p)

r, p = np.meshgrid(r, p)
Z = r*np.cos(p) # 表示する計算式の指定。等高線はZに対して作られる。

plt.pcolormesh(r, p, Z, cmap='hsv') # 等高線図の生成。cmapで色付けの規則を指定する。

pp=plt.colorbar (orientation="vertical") # カラーバーの表示
pp.set_label("Label", fontname="Arial", fontsize=24) #カラーバーのラベル

plt.xlabel('X', fontsize=12)
plt.ylabel('Y', fontsize=12)
plt.xlim(-1,1)
plt.ylim(-1,1)

plt.show()

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[8524ba23]

手書きでよいので、提示されたr,p,zをもとにした欲しい結果（図）を提示ください。

[mizumimizu]

返信ありがとうございます。
画像貼れなかったので、リンク張っておきます。
イメージリンク先のグラフです。
よろしくお願いします。

http://0fulo.blog.fc2.com/blog-entry-27.html

[8524ba23]

リンクは質問本文に追記ください。（質問は修正できます）
図のイメージは分かったのですが、依然として提示された具体的なr,p,zによって得られる欲しい結果（だいたいどんな色分布になるのか）が分かりませんので、できれば手書きで良いので提示ください。

Answer 1

自分でわかりました！
コード書いておきます。

import matplotlib.pyplot as plt
import numpy as np

r = np.linspace(0, 1.0, 200)
p = np.linspace(0, 2np.pi, 300)
R, P = np.meshgrid(r, p)
Z = Rnp.cos(P)

X, Y = Rnp.cos(P), Rnp.sin(P)

plt.pcolormesh(X,Y,Z,cmap='hsv') #等高線図の生成。cmapで色付けの規則を指定する
plt.colorbar(orientation="vertical")#カラーバーの表示

plt.xlabel('X', fontsize=12)
plt.ylabel('Y', fontsize=12)
plt.xlim(-1,1)
plt.ylim(-1,1)

plt.xticks(np.linspace(-1, 1, 5))
plt.yticks(np.linspace(-1, 1, 5))

g=plt.subplot()
g.set_aspect('equal')

plt.show()

Answer 2

提示されたリンク先には完全に動作するコードがあるので、まずはそれをそのまま実行してみましょう。
そのうえで諸元(r,p,z)をご自身のものに修正するとよいでしょう。

ただし、こちらで実行してみたところ
df.to_csv("～testdata.csv", index=None) #csvに出力
を
df.to_csv("～testdata.csv", index=None, header=None) #csvに出力
と修正する必要がありました。

諸元を適当に変えて描画してみた結果