Pandas:使用其他列的计算添加列

时间:2016-04-09 13:19:34

标签: python pandas

我有一个测量的csv:

model.put("addForm", form);

我想添加另一列,其平方根为x ^ 2 + y ^ 2, Z = SQRT(X ^ 2 + Y ^ 2)

像这样:

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

有什么想法吗?

谢谢!

1 个答案:

答案 0 :(得分:5)

对方块的结果使用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