python - key 错误 : Creating a dataframe column using square root of the sum of square of two column values

Question

我有一个包含 A 列和 B 列的数据框。我想创建第三列，它是两列平方和的平方根。

以下是我的示例数据:

A   B
10  7
10  8
9   8
10  11
13  5
3   0
12  8
12  9
11  10
10  11

下面是我尝试过的代码:

df['vib_mot_sqrt'] = df[(np.sqrt(df['X']**2) + (df['index']**2))]

KeyError: "None of [Float64Index([6.0, 6.0, 11.0, 15.0, 16.0, 33.0, 42.0, 55.0, 73.0, 87.0], dtype='float64')] are in the [columns]"

Answer 1

您能否澄清一下，您想要精确计算一个值还是每一行的值？

如果您想计算每行的值并将其添加到新列，应该这样做:

df['C']= np.sqrt(df['A']**2+df['B']**2)

输出:

    A   B   C
0   4   5   6.403124
1   3   3   4.242641
2   1   1   1.414214