Pythonにおける多次元データの画像出力について

Question

前提・実現したいこと

こんにちは。Python初心者の質問になりますが、ご了承ください。
現在、Pythonで4次元のデータを画像として出力できないか模索中です。プログラムコードは抜粋ではありますが、以下の通りです。

filters = filters.reshape(64, ysize,xsize,zsize)　#ysize = 5 , xsize = 5 , zsize = 32

	fig = plt.figure()
	fig.subplots_adjust(left=0, right=1, bottom=0, top=1, hspace=0.05, wspace=0.05)
	for i in range(64):
		ax = fig.add_subplot(8, 8, i+1, xticks=[], yticks=[])
		ax.imshow(filters[i], cmap=plt.cm.gray_r, interpolation='nearest')
	filename = args[3] #ここは、無視してください
	plt.savefig(‘STEP.png’)

filtersのイメージですが、以下に示す通り、(縦、横) = (5,5)のデータが32層重なっており、さらにその3次元ボックスが64個あるといったイメージです。

実現したいこととしては、これら一つ一つ(32×64)を全部画像化してはキリがないので、たとえばzsize = 12のような領域だけ切り出して、それを64回繰り返し行いたいのです。

最終的な出力イメージとしては、以下の通りです。

これらの問題はPythonで実現可能なのでしょうか？分かりにくくて恐縮ですが、ご教授よろしくお願いいたします。

Answer 1

numpy のスライスを使えばよいのではないでしょうか？

import numpy as np
import matplotlib.pyplot as plt

ch = 12
filters = np.random.rand(64, 5, 5, 32)
print(filters.shape)  # (64, 5, 5, 32)

target = filters[..., ch]
print(target.shape)  # (64, 5, 5)

fig = plt.figure(figsize=(16, 16))
for i, arr in enumerate(target, 1):
    ax = fig.add_subplot(8, 8, i)
    ax.imshow(arr, cmap="gray")
    ax.set_axis_off()

plt.show()