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方块的结果:

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

你也可以sum行结果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