Python DataFrame Seaborn グラフに数値を入れたカラーマップ作成

Question

DataFrameで、

0 1 2 3 4
0 11 12 15 44 55
1 66 77 54 43 44
2 55 66 77 33 32
3 32 33 33 34 45
4 45 56 44 56 66

という5×5の正方行列があったとして、
インデックスはY、ヘッダーはX、値はZです。

この正方行列を、
X Y Z
0 0 0 11
1 0 1 66
2 0 2 55
3
4
・
・

という感じに並び替えたいです。
その並び替えができれば、その後の数値の入ったカラーマップは、
https://qiita.com/yoshi65/items/90532732bf9d3875bec7
こちらを参考すればできると思いますので、
まずは、行列の入れ替え方を教えていただけないでしょうか。

meshgridを使用してもうまくいきません。

何卒、宜しくお願いいたします。

Answer 1

いわゆる、横持ちから縦持ちへの変換になります。
「pandas 縦持ち 横持ち」で検索するといろいろ見つかると思いますが、
今回はunstack()して、reset_index()するのが良いと思います。

import pandas as pd

df = pd.DataFrame(
    [[11,12,15,44,55],
     [66,77,54,43,44],
     [55,66,77,33,32],
     [32,33,33,34,45],
     [45,56,44,56,66]])

df = df.unstack().reset_index()
df.columns = ['X', 'Y', 'Z']
print(df)
#    X  Y   Z
#0   0  0  11
#1   0  1  66
#2   0  2  55
#3   0  3  32
#4   0  4  45
#5   1  0  12

追記

あとから気が付きましたが、参考ページで、カラーマップを作成するときに、縦持ちにしたものを横持ちに戻す(pivot)処理をしているので、元のデータフレームをそのままヒートマップに表示してもよいのではないかと思いました。