python - 将数据框与不可散列的列合并

Question

我想合并两个 Pandas DataFrame。如果 item 代码(例如 A、B、C、D)相同，它们的属性 a, b 必须相同，但是 b 是一个 numpy 数组或列表不可散列。

富:

item   a     b              
A      1     [2,0] 
B      1     [3,0]         
C      0     [4,0]         

酒吧:

item   a     b
A      1     [2,0]
D      0     [6,1]

这就是我想要的

code   a     b        Foo   Bar
A      1     [2,0]    1     1
B      1     [3,0]    1     0
C      0     [4,0]    1     0
D      0     [6,1]    0     1

Answer 1

您可以使用 df.merge 和 df.fillna:

out = foo.assign(Foo=1).merge(bar.assign(Bar=1), 'outer').fillna(0)
print(out)

  item  a       b  Foo  Bar
0    A  1  (2, 0)  1.0  1.0
1    B  1  (3, 0)  1.0  0.0
2    C  0  (4, 0)  1.0  0.0
3    D  0  (6, 1)  0.0  1.0

---

如果b是列表类型，可以先转成元组再合并。

foo.b = foo.b.apply(tuple)
bar.b = bar.b.apply(tuple)
out = foo.assign(Foo=1).merge(bar.assign(Bar=1), 'outer').fillna(0)
out.b = out.b.apply(list)

print(out)

  item  a       b  Foo  Bar
0    A  1  [2, 0]  1.0  1.0
1    B  1  [3, 0]  1.0  0.0
2    C  0  [4, 0]  1.0  0.0
3    D  0  [6, 1]  0.0  1.0