在熊猫中加入MultiIndex系列

时间:2015-01-13 15:01:06

标签: python pandas

我有多个具有MultiIndex的系列,我想将它们组合成一个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

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

2 个答案:

答案 0 :(得分:3)

记录在案here

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

如果需要,可以使用set_index设置索引。可替换地:

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

答案 1 :(得分:0)

从pandas 0.24.0开始,可以使用重叠的索引级别将多索引数据帧彼此合并。根据{{​​3}}:

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]
相关问题