加入/合并两个Pandas数据帧。将一个级别与另一个级别的索引相匹配

时间:2016-03-29 11:30:07

标签: python join pandas merge dataframe

我正在尝试加入两个pandas数据帧;左边的一个是多索引,右边的一个是普通的香草数据帧。我想在左侧数据帧的一个级别上加入正确数据帧的索引。例如,如果我们有以下示例:

           Age
Boys          
      Sam   21
      John  22
Girls         
      Lisa  23

      Points
John       1
Lisa       2
Sam        3

我想最终得到这个:

           Age Points
Boys                 
      Sam   21      3
      John  22      1
Girls                
      Lisa  23      2

我的解决方法如下,我只是想知道是否有更简单的方式

In[2]: import pandas as pd
In[3]: idx = pd.MultiIndex(levels=[['Boys', 'Girls', ''],['Sam', 'John', 'Lisa', '']], labels=[[0,2,2,1,2],[3,0,1,3,2]])
df1 = pd.DataFrame({'Age':['',21,22,'',23]}, index=idx)
df2 = pd.DataFrame({'Points':[1, 2, 3]}, index=['John','Lisa','Sam'])

In[4]: df1
Out[4]: 
           Age
Boys          
      Sam   21
      John  22
Girls         
      Lisa  23

In[5]: df2
Out[5]: 
      Points
John       1
Lisa       2
Sam        3

然后我写了这个循环"转换"正确的数据框,给它一个多索引,并适当重新排列值

lvl = df1.index.levels[1]
lbl = df1.index.labels[1]
y = df2.iloc[:,0].values.tolist()
z=[]
for x in [lvl[k] for k in lbl]:
    try:
        idx = df2.index.tolist().index(x)
    except ValueError as e:
        z.append('')
    else:
        z.append(y[idx])

temp=pd.DataFrame(index=df1.index)
temp['Points'] = z

我现在可以加入他们

out = df1.join(temp)
out
Out[6]: 
           Age Points
Boys                 
      Sam   21      3
      John  22      1
Girls                
      Lisa  23      2

1 个答案:

答案 0 :(得分:2)

命名索引 - 它将帮助Pandas了解如何加入数据框:

In [72]: df1
Out[72]:
           Age
sex   name
Boys
      Sam   21
      John  22
Girls
      Lisa  23

In [73]: df1.index.names=['sex','name']

In [74]: df2.index.name = 'name'

现在加入非常简单:

In [75]: df1.join(df2)
Out[75]:
           Age  Points
sex   name
Boys               NaN
      Sam   21       3
      John  22       1
Girls              NaN
      Lisa  23       2

PS NaNs - 是空行的结果