二点間の距離を求めるプログラムの自動化

Question

前提・実現したいこと

引数に変数を連番で入れてプログラムを回したいです。
やりたいことは、
x(n) = 1.0,2.0,...
y(n) = 2.0,2.0,...
として、z,wは固定で自分が入力した点(x(n),y(n))と(z,w)との距離を
求めていくプログラムを作りたいです。

該当のソースコード

import math

def get_distance(x1, y1, z, w):
    d = math.sqrt((z - x1) ** 2 + (w - y1) ** 2)
    return d

if __name__ == '__main__':
    x1 = 1.0
    y1 = 2.0
    z, w = 2.0, 3.0
    d = get_distance(x1, y1, z, w)
    print(d)

試したこと

x2,x3,x4,x5,x6=2.0,1.0,3.0,1.0,1.0
y2,y3,y4,y5,y6=2.0,4.0,1.0,2.0,3.0
のようにひとつひとつx,yの値を書いて行く方法を試しましたが、
数が100個以上あるので自動化したいです。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[tiitoi]

自動化がなにを指すのかよくわからないのですが、点の一覧をリストで書いて、for で回して計算していくやり方では駄目なのでしょうか？

Answer 1

x,yの各値をリストで持ち、forループで処理すればよいです。
なお、x,yを同時に走査できるzip関数も利用しています。
参考：forループで便利な zip, enumerate関数

import math
def get_distance(x1, y1, z, w):
    d = math.sqrt((z - x1) ** 2 + (w - y1) ** 2)
    return d

xn = [1.0,2.0,1.0,3.0,1.0,1.0]
yn = [2.0,2.0,4.0,1.0,2.0,3.0]
z,w = 2.0, 3.0
dn = [] # 結果
for x,y in zip(xn,yn):
    dn.append( get_distance(x,y,z,w))
print(dn) # [1.4142135623730951, 1.0, 1.4142135623730951, 2.23606797749979, 1.4142135623730951, 1.0]
exit(0)

Comments

[pandako07]

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

[LouiS0616]

get_distanceの実装には、math.hypotが使えそうな気がします。