Python 如何用相同索引的同一数据框中的另一行替换行

发布于06月21日

我有一个数据帧df1:

id1_a    id2_a   id3_a  id1_b    id2_b   id3_b
a1        a2       a3    b1        b2     b3
aa1       a2       a3    b12       b22    b23
ac1       c2       a3    bc2       bc2    bc3

我想将其用作字典来替换中id1\u a、id2\u a、id3\u a列中的值

此数据帧df2:

id1_a    id2_a   id3_a  
a1        a2       a3    
a1        a2       a3   
aa1       a2       a3    
ac1       c2       a3

所以期望的结果是

id1_a    id2_a   id3_a  
b1        b2       b3   
b1        b2       b3
b12       b22      b23   
bc2       bc2      bc3

我怎么能那样做？如果只有一列，我会将其转换为字典并替换值，但在三列的情况下该怎么办？

推荐答案

如果需要使用在_a到_b列之间创建的字典替换所有列，请使用:

df1.columns = df1.columns.str.split('_', expand=True)
df1 = df1.stack(0)

df2 = df2.replace(dict(zip(df1['a'], df1['b'])))
print (df2)

  id1_a id2_a id3_a
0    b1   b22   bc3
1    b1   b22   bc3
2   b12   b22   bc3
3   bc2   bc2   bc3

如果需要替换为匹配列，请创建Series以首先指定它:

print (df1)
  id1_a id2_a id3_a id1_b id2_b id3_b
0    a1    a2    a3    b1    b2    b3
1   aa1    a1    a3   b12   b22   b23
2   ac1    c2    a3   bc2   bc2   bc3


print (df2)
  id1_a id2_a id3_a
0    b1   b22   bc3
1    b1    b2   bc3
2   b12    b2   bc3
3   bc2   bc2   bc3


df1.columns = df1.columns.str.split('_', expand=True)
s = (df1.stack(0)
        .groupby(level=1)
        .apply(lambda x: dict(zip(x['a'], x['b'])))
        .add_suffix('_a'))
print (s)
id1_a    {'a1': 'b1', 'aa1': 'b12', 'ac1': 'bc2'}
id2_a      {'a2': 'b2', 'a1': 'b22', 'c2': 'bc2'}
id3_a                               {'a3': 'bc3'}
dtype: object

df2 = df2.replace(s)
print (df2)
  id1_a id2_a id3_a
0    b1   b22   bc3
1    b1    b2   bc3
2   b12    b2   bc3
3   bc2   bc2   bc3