Pythonで配列から円形にデータを抽出したい

Question

前提・実現したいこと

9×9の配列から半径4の円形にデータを抽出したいと考えています。
円は媒介変数を使い、円の中心はdeta[5,0]とします。
下記のようなイメージです。どなたかできる方がいらっしゃいましたら、ご教授お願い致します。

円の設定

import numpy as np
theta = np.linspace(0, 2*np.pi, 100) #角度
r=4.　#半径
cx,cy=Circle_func(theta,r)

def Circle_func(theta,r):
    return r*np.cos(theta),r*np.sin(theta)

該当のソースコード

np.whereを使って円より内側を判別すればよいと考えているのですが、どうすればいいかわかりません。

np.where()

Answer 1

numpy.meshgridを使って、インデックスの二次元配列を作ってやると、
それで各点の中心からの距離が計算できるので、円内かどうか判定できます。

import numpy as np

N = 9

i0, j0 = 5, 0
r = 4

ii, jj = np.meshgrid(np.arange(N), np.arange(N), indexing='ij')
d2 = np.square(ii - i0) + np.square(jj - j0)

#data = np.arange(N * N).reshape(N, N)
print(np.where(d2 < r * r, data, np.nan))

Comments

[MT7678]

ご回答ありがとうございます。

Answer 2

三平方の定理、を使えば定距離にある位置データは算出できるんで、
そこらへんから考えてみればどうでしょう

Comments

[MT7678]

ご回答ありがとうございます。確かに三平方を使えばできますね。ありがとうございます。