python csvに格納された座標（ｘ、ｙ）と点（ｘ、ｙ）の距離（２点間の距離）

Question

2点間の距離（ユークリッド距離）について質問です。

A(x1=2,y1=2),B(x2=4,y2=6)である時の２点間の距離は以下のスクリプトでできるかと思います。

import numpy as np
import pandas as pd

x1 = 2
y1 = 2
x2 = 4
y2 = 6

a = np.array([x1,y1])
b = np.array([x2,y2])

u = b - a
np.linalg.norm(u)
print(np.linalg.norm(u))

これとは別に、csvに格納されているそれぞれのⅹ,yと設定した点の２点間距離を求めるためにはどうしたらよいでしょうか。
x,y
4,3
7,8
5,9

設定した点（x1,y1)(2,2)
以下が試したスクリプトです。

x1 = 2
y1 = 2

df = pd.read_csv("x_y.csv",usecols=[0,1])

df_x = df["x"]
df_y = df["y"]

a = np.array(df_x,df_y)
b = np.array([x1,y1])

u = b - a
np.linalg.norm(u)
print(np.linalg.norm(u))

エラーメッセージ

Message=operands could not be broadcast together with shapes (2,) (3,)
Source=C:\Users\bosailab\source\repos\point_distance\point_distance\point_distance.py
スタック トレース:
File "C:\Users\bosailab\source\repos\point_distance\point_distance\point_distance.py", line 18, in <module>
u = b - a

x_y.csvに格納されたすべての(x,y)と設定した点（2,2）の距離を算出したいです。

お手数をおかけいたしますが、よろしくお願いいたします。

Comments

[hayataka2049]

x_y.csvの中身をそのまま教えて下さい（質問文にあるものはcsvフォーマットではないです）

[hayataka2049]

あと、エラーメッセージはすべて掲載してください（スタックトレース含めてすべてです）

[huuuuuuu]

hayataka2049さん>失礼いたしました。２点修正いたしました。よろしくお願いいたします。

Answer 1

こんな感じでやってください。

import numpy as np
import pandas as pd

x1 = 2
y1 = 2

df = pd.read_csv("x_y.csv",usecols=[0,1])

df_x = df["x"]
df_y = df["y"]

a = np.array([df_x,df_y]).T  # 修正点1
b = np.array([x1,y1])

u = a - b
print(np.linalg.norm(u, axis=1))  # 修正点2

Comments

[huuuuuuu]

ありがとうございます。解決致しました。
もう一つご質問なのですが、
csvファイルに『distance』という列を追加したのですが、
ここに算出した値を出力する為にはどのようにしたらよろしいでしょうか。
to_csvを用いるということはわかるのですが、どのように記述したらよいかがわかりません。
基礎的な内容で申し訳ございませんが、よろしくお願いいたします。
x,y,distance
4,3,
7,8,
5,9,

[hayataka2049]

dist = np.linalg.norm(u, axis=1)
df["distance"] = dist
df.to_csv("x_y_dist.csv", index=False)

などでどうぞ。
（別ファイルに書き出すように書いていますが、to_csvの第一引数に読み込み時と同じ名前を指定すれば上書きされます）

[huuuuuuu]

ありがとうございます。
ご教授いただいた内容で解決いたしました。