基于其他列创建一个新列作为另一个数据帧的索引

Question

假设我有一个至少有两列col1和col2的数据帧。此外，我还有另一个数据帧，其列名是col1中的值，其索引是col2中的值。

import pandas as pd
df1 = pd.DataFrame( {'col1': ['x1', 'x2', 'x2'], 'col2': ['y0', 'y1', 'y0']})
print(df1)
  col1 col2
0   x1   y0
1   x2   y1
2   x2   y0

print(df2)
     y0   y1
x1    1    4
x2    2    5
x3    3    6

现在我想添加col3，它在col1的索引处和col2的列中给出第二个数据帧的值。结果应该如下所示：

   col1  col2  col3
0    x1    y0     1
1    x2    y1     5
2    x2    y0     2

谢谢大家!！

Answer 1

Pandas在索引和列上都支持联接操作，这意味着您可以执行以下操作：

df1.merge(df2, left_on='col1', right_index=True)

生产

  col1 col2  y0  y1
0   x1   y0   1   4
1   x2   y1   2   5
2   x2   y0   2   5

下一步是将正确的值输入col3

这是一个有点低效但它是一种方法得到正确的数据到一个列

df['col3'] = df[['col2', 'y0', 'y1']].apply(lambda x: x[int(x[0][1]) + 1], axis=1)

Answer 2

您可以使用堆栈为新的df与合并：

df2 = df2.stack().reset_index()
df2.columns = ['col1','col2','col3']
print (df2)
  col1 col2  col3
0   x1   y0     1
1   x1   y1     4
2   x2   y0     2
3   x2   y1     5
4   x3   y0     3
5   x3   y1     6

print (pd.merge(df1, df2, on=['col1','col2'], how='left'))
  col1 col2  col3
0   x1   y0     1
1   x2   y1     5
2   x2   y0     2

另一个解决方案是使用join创建新的系列：

s = df2.stack().rename('col3')
print (s)
  col1 col2
0   x1   y0
1   x2   y1
2   x2   y0
x1  y0    1
    y1    4
x2  y0    2
    y1    5
x3  y0    3
    y1    6
Name: col3, dtype: int64

print (df1.join(s, on=['col1','col2']))
  col1 col2  col3
0   x1   y0     1
1   x2   y1     5
2   x2   y0     2