等高線の色を変えたい

Question

CSVから読みこんだデータから等高線を作っているのですが，色の変更が上手くいきません．エラーは出ないのですが・・・

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



# データ
p2 =np.loadtxt('initial.csv',delimiter=',')
# 軸作成
a,b = [],[]
x = p2[0,:]
y = p2[:,0]
x = y[1:]
y = x[1:]
for num in range(len(x)):
    a.append(y)
for num in range(len(y)):
    b.append(z)
c = np.array(a)
d = np.array(b).T
p2 = np.delete(p2,0,1)
p2 = np.delete(p2,0,0)
#グラフ
plt.contourf(c,d,p2,1000)
plt.xlabel('takasa')
plt.ylabel('haba')
plt.imshow(p2,cmap=cm.bwr)
plt.colorbar()
plt.show()

Answer 1

どのようなグラフを作図したいのかがいまひとつ理解できておりませんが、
単に等高線を描いて色をつけたいというのであれば

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

# データ
p2 =np.loadtxt('density_python.csv',delimiter=',')

# yz軸作成
Y,Z = np.meshgrid(p2[0, 1:], p2[1: ,0])

#データ2次元配列作成
p2 = p2[1:,1:]

#グラフ
plt.contourf(Y, Z, p2, cmap=cm.bwr_r, vmin=-100, vmax=100)
plt.xlabel('Y')
plt.ylabel('Z')
plt.colorbar()
plt.show()

で良いのではないでしょうか。

Comments

[shougi]

質問文のコードの下から3行目のplt.imshow(p2,cmap=cm.bwr)を削除すれば，ネット上にあるコードなので正常に動作するのですが，色が好みではないので，プラスが青，マイナスが赤で0に近くなるほど白くなるように変えたいのです

[shougi]

ありがとうございます

[magichan]

まず、「プラスが青，マイナスが赤」の件ですが、現状使用しております "cm.bwr" と色が反対となりますので "cm.bwr_r" をお使いください。
次にセンターを 0 付近にする方法ですが、 vmax, vmin を使用して cmap の上限と下限を設定できますので、例えば p2 の最大値が 100 の場合は vmax=100, vmin=-100 などとすることで、センターを0とすることが出来ます。

[shougi]

なるほど，ありがとうございます．p2とcmapの間に，数字を入れれば等高線も細かくできました．