在 pandas 加入 MultiIndex 系列

Question

我有多个带有 MultiIndex 的 Series，我想将它们合并到一个 DataFrame 中，该 DataFrame 将它们连接到公共索引名称（并广播值）。设置就像

import pandas as pd
a=pd.Series([12,13,14,15], index=pd.MultiIndex.from_tuples([(1,1),(1,2),(2,1),(2,2)], names=["i", "j"]))
b=pd.Series([21,22], index=pd.MultiIndex.from_tuples([(1,),(2,)], names=["i"]))

我怎样才能得到结果

i  j  a  b
1  1  12 21 
1  2  13 21
2  1  14 22
2  2  15 22

你能建议如何得到这个结果吗？理想情况下，这也应该适用于两个以上的系列。

Answer 1

这已记录在案 here:

pd.merge(a.reset_index(), b.reset_index(), on='i')

如果需要，您可以使用 set_index 设置索引。或者：

pd.DataFrame(b).join(pd.DataFrame(a), rsuffix='a')

Answer 2

从 pandas 0.24.0 开始，可以使用重叠索引级别将多索引数据帧相互合并。根据 the release notes:

index_left = pd.MultiIndex.from_tuples([('K0', 'X0'), ('K0', 'X1'),
                                        ('K1', 'X2')],
                                        names=['key', 'X'])

left = pd.DataFrame({'A': ['A0', 'A1', 'A2'],
                     'B': ['B0', 'B1', 'B2']}, index=index_left)

index_right = pd.MultiIndex.from_tuples([('K0', 'Y0'), ('K1', 'Y1'),
                                        ('K2', 'Y2'), ('K2', 'Y3')],
                                        names=['key', 'Y'])

right = pd.DataFrame({'C': ['C0', 'C1', 'C2', 'C3'],
                      'D': ['D0', 'D1', 'D2', 'D3']}, index=index_right)

left.join(right)

输出：

            A   B   C   D
key X  Y                 
K0  X0 Y0  A0  B0  C0  D0
    X1 Y0  A1  B1  C0  D0
K1  X2 Y1  A2  B2  C1  D1

[3 rows x 4 columns]

在 pandas 加入 MultiIndex 系列

Join Series on MultiIndex in pandas

python

pandas