按值比较数据帧

时间:2016-06-07 10:02:38

标签: python pandas

我有两个带有几何测量值的文件: 每个文件包含大量的点(由x,y和z定义)

我想比较水平坐标相同(df1['x'] == df2['x']) & (df1['y'] == df2['y'])的深度(z值)。然而,水平坐标的顺序在两个文件中可以不同。示例数据:

x1   y1   z1
1    1    1.5
1    2    0.5
2    1    0.2
2    2    2.2
3    1    0.6
3    2    0.7

x2   y2   z2
2    1    0.5
2    2    0.2
2    3    0.5
1    1    0.1
1    2    2.3
1    3    1.7

我想要一个像这样的结果数据帧(z1 - z2,其中x和y匹配):

x    y    z
1    1    1.4
1    2    -1.8
2    1    -0.3
2    2    2.0

我将如何做到这一点?

1 个答案:

答案 0 :(得分:1)

您可以merge然后减去:

In [67]:
merged = df.merge(df1, left_on=['x1','y1'], right_on=['x2','y2'])
merged['z'] = merged['z1'] - merged['z2']
merged.drop(['x2','y2','z1','z2'],axis=1,inplace=True)
merged = merged.rename(columns={'x1':'x','y1':'y'})
merged

Out[67]:
   x  y    z
0  1  1  1.4
1  1  2 -1.8
2  2  1 -0.3
3  2  2  2.0

如果你merge它将执行inner类型合并,只有在值匹配并删除所有无关行的位置时,您可以执行减法,然后删除/重命名列以获得所需的结果