如何在分层索引的DataFrame中的单个索引上进行广播?

时间:2012-09-17 14:29:46

标签: python pandas

我有一个分层索引的数据框:

>>> import pandas as pd
>>> df = pd.DataFrame(np.arange(4),
                      index=[['John', 'John', 'Vicki', 'Vicki'], 
                             ['a','b', 'a','b']],
                      columns=['score'])

         score
John  a      0
      b      1
Vicki a      2
      b      3

和索引与上述数据框中第一个索引级别相同的系列:

>>> series = pd.Series([100, 200], index=['John', 'Vicki'])

John     100
Vicki    200

现在我想将数据框与系列合并,以便沿着第二级索引广播系列中的值。结果数据框应如下所示:

         score  salary
John  a      0     100
      b      1     100
Vicki a      2     200
      b      3     200

我怎样才能在熊猫中实现这一目标?

1 个答案:

答案 0 :(得分:2)

这应该有效:

df['salary'] = series.reindex(df.index, level=0)