python pandas 合并_python – Pandas合并并保留索引

在

How to keep index when using pandas merge中提出了类似的问题,但它不适用于MultiIndexes,即

a = DataFrame(np.array([1,2,3,4,1,2,3,3]).reshape((4,2)), columns=['col1','to_merge_on'], index=['a','b','a','b'])

id = pd.MultiIndex.from_arrays([[1,1,2,2],['a','b','a','b']], names =['id1','id2'])

a.index = id

In [207]: a

Out[207]:

col1 to_merge_on

id1 id2

1 a 1 2

b 3 4

2 a 1 2

b 3 4

b=DataFrame(data={"col2": [1,2,3], 'to_merge_on' : [1,3,5]})

In [209]: b

Out[209]:

col2 to_merge_on

0 1 1

1 2 3

2 3 5

a.reset_index().merge(b, how="left").set_index('index')

In [208]: a.reset_index().merge(b, how="left").set_index('index')

------------------------------------