python pcolormesh での各軸の要素数

Question

Matplotlibのpcolormeshを使って2次元ヒートマップを作ろうとしています。

次のようなエラーが出ていますが、要素数は合っているように思います。

TypeError: Dimensions of C (6, 3000) are incompatible with X (3001) and/or Y (7); see help(pcolormesh)

Cの配列を転置しても同様なエラーが発生します。
私の理解ではpcolormeshは塗りつぶすグリッドの端の値を指定するため、X,Yの要素数は
それぞれ＋１必要だと思っていますが、間違っていますでしょうか。

Spyder4.1.3で実行しています。

簡単な内容ですみません。
コードなど必要であれば載せますので、お申し付けください。

よろしくお願いします。

Answer 1

shadingを'gouraud'にしていたのですが，'flat'とすると, 動作しました.

shading の指定の話でしたか。その場合、"flat" と "gouraud" で引数 C の解釈が異なってきます。

pcolormesh(X, Y, C) の指定ですが、

shading="flat" の場合は、C はそのまま各メッシュの色に対応づけられるので、X, Y が (M, N) の2次元配列であれば、C は (M - 1, N - 1) の2次元配列を指定します。

一方、shading="gouraud" の場合、C は各メッシュの色ではなく、X, Y の各点の色と対応づけられるので、X, Y が (M, N) の2次元配列であれば、C も (M, N) の2次元配列を指定します。
実際のメッシュの色は、グーローシェーディング によって、ポリゴンの頂点の色から補間されて生成されます。

matplotlib - pcolormesh でメッシュを描画する方法

Comments

[drug24]

ご丁寧に回答して頂き、ありがとうございました。
質問文に記載が無い部分ですが、

shading='flat'

とすると描画できました。
本当はshading='gouraud'
の状態で描画したかったのですが...

[tiitoi]

コードがないので、状況がわかりません。
コピペで動作するコードを追記してください

[drug24]

pcolormeshのパラメータの1つの
shadingを'gouraud'にしていたのですが，
'flat'とすると, 動作しました.

Cの各軸の要素数とX,Yの要素数が等しい場合は
shading='goround'
でも問題なく動いたので，内挿するときの計算が関係しているのでしょうか.

[tiitoi]

追記しました。shading='goround' の場合、C は各メッシュの色ではなく、X、Y の点の色なので、X、Y と同じ形状でないといけません。
メッシュの色は各頂点の色からグーローシェーディングで補間されます。