python - 2个数据框之间的复杂链接

Question

我有 2 个数据框。第一个是我的参考

df_ref
ID  REF  VALUE
A   1    12
A   2    36
A   3    95
B   1    54
B   2    67
B   3    81
C   1    89
C   2    123
C   3    14

第二个是我的限制：

df_restrict
ID  V1  V2
A   1   2
B   3   2
C   2   1

我想要的是用df_restrict对应于(ID, REF)from的值替换 V1 和 V2 indf_ref

df_restrict
ID  V1  V2
A   12  36
B   81  67
C   123 89

提前致谢

标签： pythonpython-3.xpandasdataframe

Answer 1

我们可以旋转df_ref到重塑，然后将索引设置df_restrict为ID，然后转置并替换重塑后的值df_ref

r = df_ref.pivot('REF', 'ID', 'VALUE')
df_restrict.set_index('ID').T.replace(r).T

---

     V1  V2
ID         
A    12  36
B    81  67
C   123  89