ループで計算されないため解決方法が知りたいです

Question

ランダムで生成した3つのハブ駅とランダムで生成した25駅のリストを用意し、その後3つのハブ駅のある一定半径円内に含まれている駅を除いたリストを生成したいのですが、最後の一個の要素でしか計算されません。

初心者で初歩的なところでつまずいているのはわかっているのですが、どなたか教えていただけるとありがたいです。
よろしくお願いいたします。

def remove_station(station_list, hub_list, R_big):
    X, Y = hub_list[0], hub_list[1]
    x, y = station_list[0], station_list[1]
    hub_size = len(hub_list[0])
    station_size =len(station_list[0])
    
    for i in range(hub_size):
        for j in range(station_size):
            Dx = X[i] - x[j]
            Dy = Y[i] - y[j]
            b = math.sqrt(Dx * Dx + Dy * Dy)
    
    m = []
    n = []
    if b - R_big >= 0: # 駅と拠点駅の距離がR_big以上であれば、駅として採用
        m.append(x[j])
        n.append(y[j])
    else:
        pass

    return [m, n]

Comments

[tiitoi]

remove_station の呼び出し例 (どういう引数が渡されるのか) を追記していただけますか。

[退会済みユーザー]

説明不足で失礼いたしました。

引数としては
remove_station(make_station(station_number, station_upper), hub_station(hub_number, hub_upper), R_big)
で、中身としては、

def make_station(station_number, station_upper):
random.seed(1)
x = [random.randint(0, station_upper) for i in range(station_number)]
y = [random.randint(0, station_upper) for i in range(station_number)]
return [x, y]

def hub_station(hub_number, hub_upper):
random.seed(1)
X = [random.randint(0, hub_upper) for i in range(hub_number)]
Y = [random.randint(0, hub_upper) for i in range(hub_number)]
return [X, Y]

です。

Answer 1

こんにちわ。

二重ループを抜けた後でリストm,nが初期化されているため、最後の1件しかm,nには入ってきません。
m,nの初期化を行うのは2重ループの前ですね。

ある駅とすべてのハブ駅との間の距離がR_big以上離れているかどうか判定する必要が
あるので、二重ループは普通の駅のリストが外でhub駅のリストが中ですね。

あと、プログラムが正確に書けているかわからない状態でランダムなデータを使うと、
プログラムの検証が難しくなりますよ。

その点を踏まえて修正してみました。以下をご覧ください。

ハブ駅の座標は以下の通りです。
(25,25)
(75,25)
(75,75)

普通の駅の座標は以下の通りです。
(25,20)
(75,30)
(75,60)
(25,75)
(25,60)

半径R_bigは25としました。

絵を書いてみればわかるのですが、(25,75),(25,60)の2駅がどのハブ駅からも離れています。

import math
def remove_station(station_list, hub_list, R_big):
    X, Y = hub_list[0], hub_list[1]
    x, y = station_list[0], station_list[1]
    hub_size = len(hub_list[0])
    station_size = len(station_list[0])

    m = []
    n = []
    for j in range(station_size):
        flag = True
        for i in range(hub_size):
            Dx = X[i] - x[j]
            Dy = Y[i] - y[j]
            b = math.sqrt(Dx * Dx + Dy * Dy)
            if b <= R_big:
				# あるハブ駅と駅との距離がR_big以内であれば、
				# 駅として採用しないのでflagにFalseをセットする
                flag = False
        if flag == True:
            # どのハブ駅からもR_big以上の距離がある駅を
            # リストm,nに追加する
            m.append(x[j])
            n.append(y[j])
            
    return [m, n]

(station_list_x,station_list_y) = remove_station([[25,75,75,25,25],[20,30,60,75,60]],[[25,75,75],[25,25,75]],25)
print('x = ')
print(station_list_x)
print('y = ')
print(station_list_y)

実行結果は以下の通りです。

x = 
[25, 25]
y = 
[75, 60]

参考になれば幸いです。

Comments

[退会済みユーザー]

丁寧な回答ありがとうございました。
大変参考になりました！
ランダムを使うのは確かに確認するときに不便ですね。以後気をつけます！

Answer 2

ざっと一瞬見た感想。

if文がfor文の中にないため、うまくいかないだけと思います。