Q&A
(h*w*3) という1次元配列でなく、(H, W, 3) という3次元配列として扱えばよいのではないでしょうか？
numpyの練習として，画像をrgbからhsvへ変換するコードを書いていました．
rgbですので画像のはshapeは，高さh，幅wとすると，hw3です．
rgbからhsvへ変換する際にhを求める式では，
最大値が
rの場合 g-b
gの場合 b-r
bの場合 r-g
とそれぞれ最大値以外の値を用いた計算が必要となります．（参考url）
numpyの操作として最大値を取り出すのは造作もないのですが，最大値の右隣や左隣の要素を取り出すのが難しいです．
取り出すべき値のindexについては，argmaxを用いて得られるために得たindexに+1もしくは+2をした後に%3と余りを求めれば隣の要素のindexも得られるために，内包表記を用いてTrue Falseで位置を示す配列を作成することは出来ました．
しかしそれはスマートではありません．
何か良い方法はないでしょうか．
ん、shape=(h*w*3,)なんですかね。勝手に(H, W, 3) だと思ってたけど。
回答1件
0
ベストアンサー
とりあえずこうなる。
python
1>>> import numpy as np 2>>> a = np.array([[[0,1,2],[3,5,4]],[[8,6,7],[9,11,10]]]) 3>>> a 4array([[[ 0, 1, 2], 5 [ 3, 5, 4]], 6 7 [[ 8, 6, 7], 8 [ 9, 11, 10]]]) 9>>> argmax = a.argmax(axis=2) 10>>> argmax 11array([[2, 1], 12 [0, 1]]) 13>>> idx1 = (argmax+1)%3 14>>> idx2 = (argmax+2)%3 15>>> idx1 16array([[0, 2], 17 [1, 2]]) 18>>> idx2 19array([[1, 0], 20 [2, 0]])
ここから先ですが、私はこの間こんな回答をしたので、この考え方に基づいてやることにします。
Python - Numpy 3Darrayのインデックスを2Darrayで指定したい(再度)｜teratail
python
1>>> (np.take_along_axis(a, idx1.reshape(idx1.shape+(1,)), axis=2) - np.take_along_axis(a, idx2.reshape(idx2.shape+(1,)), axis=2)).reshape(idx1.shape) 2array([[-1, 1], 3 [-1, 1]])
少し改良するとこう。
python
1>>> tmp = np.take_along_axis(a, np.stack([idx1, idx2], axis=2), axis=2) 2>>> tmp[:,:,0] - tmp[:,:,1] 3array([[-1, 1], 4 [-1, 1]]) 5
投稿2019/02/08 16:02
総合スコア30933
あなたの回答
tips
プレビュー
