python - Pandas 如何根据两列合并两个数据框并更新下一列？

Question

我想合并两个 pandas 数据框，并希望基于两列合并它们。

这是代码:

import numpy as np
import pandas as pd
from itertools import product

# first dataframe
data = np.array([[1, 1, 6],
       [1, 2, 2],
       [2, 1, 1],
       [2, 2, 9],
       [3, 1, 1],
       [3, 2, 1]])

df = pd.DataFrame(data, columns=['x','y','z'])
print(df)
   x  y  z
0  1  1  6
1  1  2  2
2  2  1  1
3  2  2  9
4  3  1  1
5  3  2  1

## another dataframe
N = 4
perms = list(product(range(N), range(N)))
x = [i[0] for i in perms]
y = [i[1] for i in perms]
df1 = pd.DataFrame({'x': x,'y': y,'z':0})

print(df1)

输出应如下所示:

    x  y  z  
0   0  0  0  
1   0  1  0  
2   0  2  0  
3   0  3  0  
4   1  0  0  
5   1  1  6  
6   1  2  2  
7   1  3  0  
8   2  0  0  
9   2  1  1 
10  2  2  9  
11  2  3  0  
12  3  0  0  
13  3  1  1  
14  3  2  1  
15  3  3  0

问题如何从数据帧 df 更新 df1 z 列的值？

Answer 1

首先合并，然后更新

df1.update(df1.drop('z',1).merge(df,how='left'))
df1
Out[439]: 
    x  y    z
0   0  0  0.0
1   0  1  0.0
2   0  2  0.0
3   0  3  0.0
4   1  0  0.0
5   1  1  6.0
6   1  2  2.0
7   1  3  0.0
8   2  0  0.0
9   2  1  1.0
10  2  2  9.0
11  2  3  0.0
12  3  0  0.0
13  3  1  1.0
14  3  2  1.0
15  3  3  0.0