python - Euclidean distance of points coordinates in pandas dataframe

Question

I have a pandas dataframe with six columns, first three columns contain x, y and z reference coordinate, and the next three - coordinates of some point. I want to put euclidean distance between those two points in new column of the dataframe. I think about using numpy.linalg.norm via pandas.apply method, but I don't know what is the best method to parse dataframe row for numpy function. Could you give me some suggestions?

Answer 1

You might not need any fancy magic here:

df['dist'] = np.sqrt( (df.x1-df.x2)**2 + (df.y1-df.y2)**2 + (df.z1-df.z2)**2)