我正在尝试加入两个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
答案 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 - 是空行的结果