python コンター Z軸設定

Question

pythonのコンターZ軸の設定について、ご教示いただけますと幸いです。

CSVファイルの3次元データ(添付図1)の可視化を行っています。
(A列にX軸,1行目にY軸,その他に実験値(Z軸)が保存されているデータです)

グラフ化の際、X軸,y軸の設定はplt.xlim(),plt.ylim()で設定できるのですが、
実験値(Z軸)の範囲設定方法が分からず、ご知見賜れますと幸いです。

実験値の最大値が3.16の場合、自動的にZ軸のカラーバーは03.2で表示されてしまいます(添付図2)。
例としてZ軸を02.0で表示したい場合(イメージでいうとplt.zlim(0,2.0))どのような方法があるか、
ご存知でしたら、ご教示いただけますと幸いです。

import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
from mpl_toolkits.mplot3d import Axes3D

p2 = np.loadtxt('test.csv',delimiter=',')

yy,zz = [],[]
y = p2[0,:]#全ての列の0行目を取得(10,11,,,5000)
z = p2[:,0]#全ての行の0列目を取得(0,1,2,,,)
y = y[1:]
z = z[1:]

for num in range(len(z)):
    yy.append(y)
for num in range(len(y)):
    zz.append(z)
X = np.array(yy)
Y = np.array(zz).T

p2 = np.delete(p2,0,1)
p2 = np.delete(p2,0,0)

plt.contourf(Y,X,p2,cmap="rainbow")
plt.colorbar()
plt.show()
plt.xlim([0,250])
plt.ylim([0,200])

jbpb0様から頂いた結果を追記しました。(図3)

変更前↓

plt.contourf(Y,X,p2,cmap="rainbow")

にvminとvmaxを追記しましたが、(下記参照)　変更後↓

plt.contourf(Y,X,p2,cmap="rainbow", vmin=0, vmax=2)

下図のようにカラーバーの最大値(3.2)は変わらず、カラーバー内の色のみかわる結果となりました。

欲しい図のイメージとしては下図の右図のように
グラフに表示するZの値をxlim,ylimのように変更した状態でして、
もしご存知の方いらっしゃいましたらご教示いただけますと幸いです。
よろしくお願いいたします。
jbpb0様ご教示いただきありがとうございました。

3/2追記
jbpb0様のご回答を頂き、levels=np.linspace(min,max,分割数に相当?)を追記することで欲しい図を出力できるようになりました。ありがとうございました。
変更した点
・ご回答の一部を追加
・subplotで複数表示 test1.csvの最大値3.18(試験結果A),test2.csvの最大値は2.0(試験結果B)

変更後のコード

import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
from mpl_toolkits.mplot3d import Axes3D

p2 = np.loadtxt('test1.csv',delimiter=',')

yy,zz = [],[]
y = p2[0,:]#全ての列の0行目を取得(10,11,,,5000)
z = p2[:,0]#全ての行の0列目を取得(0,1,2,,,)
y = y[1:]
z = z[1:]

for num in range(len(z)):
    yy.append(y)
for num in range(len(y)):
    zz.append(z)
X = np.array(yy)
Y = np.array(zz).T

p2 = np.delete(p2,0,1)
p2 = np.delete(p2,0,0)


plt.subplot(1,2,1)
plt.contourf(Y,X,p2,cmap="rainbow",levels=np.linspace(0,3.2,11))
plt.colorbar()
plt.show()
plt.xlim([0,100])
plt.ylim([0,100])

p2 = np.loadtxt('test2.csv',delimiter=',')

yy,zz = [],[]
y = p2[0,:]#全ての列の0行目を取得(10,11,,,5000)
z = p2[:,0]#全ての行の0列目を取得(0,1,2,,,)
y = y[1:]
z = z[1:]

for num in range(len(z)):
    yy.append(y)
for num in range(len(y)):
    zz.append(z)
X = np.array(yy)
Y = np.array(zz).T

p2 = np.delete(p2,0,1)
p2 = np.delete(p2,0,0)

plt.subplot(1,2,2)
plt.contourf(Y,X,p2,cmap="rainbow",levels=np.linspace(0,3.2,11))
plt.colorbar()
plt.show()
plt.xlim([0,100])
plt.ylim([0,100])

↓得られた絵

2つの試験結果の最大値を合わせて表示することができるようになり、
比較することでできるようになりました。ありがとうございました。

Comments

[jbpb0]

https://ebcrpa.jamstec.go.jp/~yyousuke/matplotlib/matplotlib.html#graph_9_4
の「陰影を付ける範囲を変更する」の方法で、いかがでしょうか？

[scirocco]

ご教示いただきありがとうございます。試してみます。

[scirocco]

ご教示大変ありがとうございました。
試してみましたが、完成イメージと異なる結果となりました。結果を追記しましたので、ご参照いただければ何卒幸いです。

[jbpb0]

> 結果を追記しました

どこに？

[scirocco]

申し訳ないです、うまく追記できておりませんでした。
ご参照いただければ幸いです。

Answer 1

「やりたいこと：」の図では、左の図と右の図で、同じ色でも数値が違います
そのようにするには、下記のようにすればできます

plt.contourf(Y,X,p2,cmap="rainbow")

↓ 変更

plt.contourf(Y,X,p2,cmap="rainbow", levels=np.linspace(0,2,11), extend='both')

そうではなく、色と数値の関係を変えずに、カラーバーに表示される範囲だけ変えたい場合は、下記のようにすればできます

plt.contourf(Y,X,p2,cmap="rainbow")
plt.colorbar()

↓ 変更

plt.contourf(Y,X,p2,cmap="rainbow")
cbar = plt.colorbar()
cbar.ax.set_ylim(0,2)

Comments

[jbpb0]

上記の二つ目の変更例で、さらに

plt.contourf(Y,X,p2,cmap="rainbow")
↓ 変更
plt.contourf(Y,X,p2,cmap="rainbow", levels=np.linspace(0,3.2,17))

とすれば、色の範囲を変えずに、色の段階を細かくできます