熊猫映射数据帧列

Question

我有两个数据帧，第一个作为值之间的成对连接：

df1 = pd.DataFrame({'n1': [5,1,1,3,4,3,2,2],
                    'n2': [1,6,3,4,3,2,3,7]})
   n1  n2
0   5   1
1   1   6
2   1   3
3   3   4
4   4   3
5   3   2
6   2   3
7   2   7

第二个表示哪些值属于每个组g为：

df2 = pd.DataFrame({'n': [1,5,6,2,3,4,7,7],
                        'g': ['a','a','a','b','b','b','c','c']})
   g  n
0  a  1
1  a  5
2  a  6
3  b  2
4  b  3
5  b  4
6  c  7
7  c  7 

我正在尝试映射数据帧以获取：

   n1  n2  g1  g2
0   5   1   a   a   
1   1   6   a   a
2   1   3   a   b
3   3   4   b   b
4   4   3   b   b
5   3   2   b   b
6   2   3   b   b
7   2   7   b   c

因此，对于每个n1和n2，创建两列，其中包含每个值所属的df2中的相应组。

到目前为止，我尝试使用以下内容进行映射：

df1['g1'] = df1['n1'].map(df2['g'])
df1['g2'] = df1['n2'].map(df2['g'])

但实际上这返回：

   n1  n2 g1 g2
0   5   1  b  a
1   1   6  a  c
2   1   3  a  b
3   3   4  b  b
4   4   3  b  b
5   3   2  b  a
6   2   3  a  b
7   2   7  a  c

因为它是在df2.index而不是n映射到g对。将df2的索引设置为g：

df2.index = df2['g']

导致以下错误：

无效索引错误：重新索引仅对唯一值索引有效 对象

Answer 1

将重复项放入df2，然后可以调用map：

In [58]:
df2 = df2.drop_duplicates()
df2
Out[58]:
   g  n
0  a  1
1  a  5
2  a  6
3  b  2
4  b  3
5  b  4
6  c  7
In [61]:
df1[['g1','g2']] = df1.apply(lambda x: x.map(df2.set_index('n')['g']))
df1
Out[61]:
   n1  n2 g1 g2
0   5   1  a  a
1   1   6  a  a
2   1   3  a  b
3   3   4  b  b
4   4   3  b  b
5   3   2  b  b
6   2   3  b  b
7   2   7  b  c