pythonでcsvデータを用いた２点間の距離の計算

Question

pythonでcsvデータ（２点のx,y座標データ）を用いて２点間の距離の計算を行いたいと考えております。

以下がスクリプトとなります。

import pandas as pd
import numpy as np

df = pd.read_csv("route_node.result.csv",usecols=[1,2,4,5])
new_file = "route_node.dist.csv"

df_x1 = df["x"]
df_y1 = df["y"]

df_x2 = df["X"]
df_y2 = df["Y"]

a = np.array([df_x1,df_y1]) 
b = np.array([df_x2,df_y2])

u = a - b
dist = np.linalg.norm(u, axis=1)
df["distance"] = dist
df.to_csv(new_file, index=False)

csvデータの中身は
start_node,x,y,route_node,X,Y
19804,-11096.6689,-43783.1815,19780,-11098.4839,-43812.357
19804,-11096.6689,-43783.1815,19798,-11097.0395,-43812.5016

のようなデータが5000行ほどあります。

上記のスクリプトを実行すると、
df["distance"] = dist　
の部分にLength of values does not match length of indexといったエラーが表示されてしまいます。
どのように修正したらよろしいでしょうか。

ご教授いただけましたら幸いです。
よろしくお願いいたします。

Answer 1

a = np.array([df_x1,df_y1]).T 
b = np.array([df_x2,df_y2]).T

とすればよいと思います。

Comments

[退会済みユーザー]

ありがとうございます。
解決いたしました。