格子点の外枠だけを求めるアルゴリズム

Question

pythonを使っています。

h, w = 30,30
step = 5
y, x = np.mgrid[step:h:step, step:w:step].reshape(2,-1)

とすると、5 から 25 まで 5 刻みの格子点のx座標とy座標が、x,yに格納されます。つまり、

(y,x) = (5,5),(5,10),(5,15),(5,20),(5,25),(10,5),…,(25,25)

といった具合です。実際に

print(y)

print(x)

とすると

[ 5 5 5 5 5 10 10 10 10 10 15 15 15 15 15 20 20 20 20 20 25 25 25 25 25]
[ 5 10 15 20 25 5 10 15 20 25 5 10 15 20 25 5 10 15 20 25 5 10 15 20 25]

と表示されます。ちなみに、ソースコード中でy,xをx,yと書かないのはわざとです。この格子点の外側だけ、

(y,x)　= (5,5),(5,10),(5,15),(5,20),(5,25),(10,5),(10,25),(15,5),(15,25),(20,5),(20,25),(25,5),(25,10),(25,15),(25,20),(25,25)

をとりだすにはどしたらよいでしょう。つまり、

print(y)

print(x)

としたときに、

[ 5 5 5 5 5 10 10 15 15 20 20 25 25 25 25 25]
[ 5 10 15 20 25 5 25 5 25 5 25 5 10 15 20 25]

と表示されるように、yとxに数字を格納したいです。スマートな方法はないでしょうか。

Answer 1

xが左辺または右辺 または yが上辺または下辺という条件で抽出すればよさそうです。
もちろん、h,w,step値およびメッシュ生成されることが分かっていれば、それらだけで計算でも求められますが。

import numpy as np
h, w = 30,30
step = 5
y, x = np.mgrid[step:h:step, step:w:step].reshape(2,-1)
left,top,right,bottom = min(x),min(y),max(x),max(y)
for yy,xx in zip(y,x):
    # xが左辺または右辺 または yが上辺または下辺
    if xx == left or xx == right or yy == top or yy == bottom:
        print(yy,xx) # 別のnumpy配列に入れるなりする

Comments

[physics303]

実装できました！
ありがとうございます！