python - Pandas:添加包含来自其他列的计算的列

Question

我有一个带有测量值的 csv:

YY-MO-DD HH-MI-SS_SSS    |        x          |          y
2015-12-07 20:51:06:608  |        2          |          4
2015-12-07 20:51:07:609  |        3          |          4

我想添加另一列，其中包含 x^2+y^2 之和的平方根，z=sqrt(x^2+y^2)

像这样:

 YY-MO-DD HH-MI-SS_SSS       |        x          |          y     |     z
    2015-12-07 20:51:06:608  |        2          |          4     |   4.472
    2015-12-07 20:51:07:609  |        3          |          4     |    5

有什么想法吗？

谢谢!

Answer 1

使用np.sqrt关于方 block 的结果:

In [10]:
df['z'] = np.sqrt(df['x']**2 + df['y']**2)
df

Out[10]:
   x  y         z
0  2  4  4.472136
1  3  4  5.000000

您还可以对 np.square 的结果按行求和 并调用 np.sqrt:

In [13]:
df['z'] = np.sqrt(np.square(df[['x','y']]).sum(axis=1))
df

Out[13]:
   x  y         z
0  2  4  4.472136
1  3  4  5.000000