python - 在数据框中操作数组数据的最佳方法？

Question

我有一个存储位置数据的数据框，如下所示：

    r1_x        r1_y        r1_z        r2_x        r2_y        r2_z
0   17.670965   19.857307   27.178185   17.555960   20.653801   27.641266
1   17.670965   19.857307   27.178185   16.950911   4.216056    12.901429
2   17.670965   19.857307   27.178185   11.737829   26.097181   3.230895
3   17.670965   19.857307   27.178185   16.254911   12.062716   12.170364
4   17.670965   19.857307   27.178185   16.504757   11.673612   12.406663
5   17.670965   19.857307   27.178185   16.717722   27.935490   28.795826

这里 x、y 和 z 表示 3d 向量的分量。在这样的数据框元素上执行向量操作的最佳方法是什么？例如，如果我想计算每一行的 r2-r1，或者取一个点积怎么办？我正在使用熊猫来存储这些数据。

标签： pythonpandasdataframe

Answer 1

尝试这个

df['r2_x-r1_x'] = df.r2_x - df.r1_x