Pythonで多次元リストから２次元ヒストグラムを作りたい

Question

Pythonで、多次元のリストからある２つの次元を指定（他の次元を固定）して、２次元ヒストグラム（離散的なヒートマップ）を作りたいです。

例えば次のような３次元のリストがあるとします。

test = [
    [
        [10, 11, 12, 13, 14],
        [11, 12, 13, 14, 15],
        [12, 13, 14, 15, 16],
        [13, 14, 15, 16, 17]
    ],
    [
        [17, 16, 15, 14, 13],
        [16, 15, 14, 13, 12],
        [15, 14, 13, 12, 11],
        [14, 13, 12, 11, 10]
    ],
    [
        [10, 11, 12, 13, 14],
        [10, 11, 12, 13, 14],
        [10, 11, 12, 13, 14],
        [10, 11, 12, 13, 14]
    ]
]

test[0][:][:], test[1][:][:], test[2][:][:]をヒストグラムとして可視化すると順に次のようになります。

これらを出力したPythonのコードは以下の通りです。

import seaborn as sns
import matplotlib.pyplot as plt

test = [
    [
        [10, 11, 12, 13, 14],
        [11, 12, 13, 14, 15],
        [12, 13, 14, 15, 16],
        [13, 14, 15, 16, 17]
    ],
    [
        [17, 16, 15, 14, 13],
        [16, 15, 14, 13, 12],
        [15, 14, 13, 12, 11],
        [14, 13, 12, 11, 10]
    ],
    [
        [10, 11, 12, 13, 14],
        [10, 11, 12, 13, 14],
        [10, 11, 12, 13, 14],
        [10, 11, 12, 13, 14]
    ]
]

plt.figure()
sns.heatmap(test[0][:][:], cmap='Blues', square=True, annot=True)
plt.savefig('./test_0.png')
plt.show()
plt.close('all')

質問

ここで３つの行列のうち、例えばそれぞれの一番上の行を取り出したもののヒストグラムを作りたいです。つまりイメージとしては

[
  [10, 11, 12, 13, 14],
  [17, 16, 15, 14, 13],
  [10, 11, 12, 13, 14]
]

をそのまま可視化したようなヒストグラムを作りたいです。これをやろうとソースコードのtest[0][:][:]をtest[:][0][:]に変えてみたのですが、ヒストグラムはtest[0][:][:]と同じものでした。同様にtest[:][1][:]はtest[1][:][:]と同じヒストグラム、test[:][2][:]はtest[2][:][:]と同じヒストグラムとなってしまいます。

どうしたらtest[:][0][:]とかtest[:][:][0]のヒストグラムを作れるでしょうか？　よろしくお願いします。

Answer 1

[

[10, 11, 12, 13, 14],
[17, 16, 15, 14, 13],
[10, 11, 12, 13, 14]
]
をそのまま可視化したようなヒストグラムを作りたいです。

下記で出来るかと思います。

sns.heatmap([test[0][0],test[1][0],test[2][0]], cmap='Blues', square=True, annot=True)

Comments

[BOLTE]

なるほど。列に関してはどのようにすればいいでしょうか？

Answer 2

回答ありがとうございます。列については次のようにして実現できました。

column_number = 2

temp_list = []
test_column = []
for first_index in range(0, 3, 1):
    temp_list_small = np.array(test[first_index]).T.tolist()
    temp_list.append(temp_list_small)
    test_column.append(temp_list[first_index][column_number])

print(test_column)

plt.figure()
sns.heatmap(test_column, cmap='Blues', square=True, annot=True)
# plt.savefig('./test_0.png')
plt.show()
plt.close('all')

temp_listはもとのリストの３つの行列を転置して配置したようなリストです。回答していただいたことをヒントに、このtemp_listを元にしてtest_columnというリストを作成しました。出力は

[[12, 13, 14, 15], [15, 14, 13, 12], [12, 12, 12, 12]]

となり、一応はできます。しかしこれだとかなりワークアラウンドな感じがします。もっとスマートに、簡単にできないでしょうか…