np.whereでドツボにハマった話 - Qiita

はじめに 画像処理の中でnp.whereのインデックス取り出しにおける出力が全くできなかったので ここのサンプルコードを参考にしながら理解しました・・・ なので備忘録です！ これ読み方わかったらアハ体験できるのでは！？？ np.whereの出力！！？ 参考にしたサンプルコードと出力です． 今回は3階テンソル($Width \times Height \times Channels = 2 \times 3 \times 4 $) となっています． a_3d = np.arange(24).reshape(2, 3, 4) print(a_3d) # output # [[[ 0 1 2 3] # [ 4 5 6 7] # [ 8 9 10 11]] # # [[12 13 14 15] # [16 17 18 19] # [20 21 22 23]]] print(np.where(a_3