PadasのMultiIndexカラムのDataFrameとシングルカラムのDataFrameを結合したい

Question

PadasのMultiIndexカラムのDataFrameとシングルカラムのDataFrameをMultiIndexのまま結合したいです。
結合した時にタプルになってしまいます。
シングルカラムをカラムのMの行かWの行に並べたいです。

import pandas as pd
import numpy as np
columns=pd.MultiIndex.from_product([['M1','M2','M3'], ['W1','W2','W3','W4','W5']],
names=['month','week'])

data=np.round(np.random.randn(4,15),1)
multidata=pd.DataFrame(data,columns=columns)
print(multidata)

test=pd.DataFrame(data=np.round(np.random.randn(4,3),1),index=[0,1,2,3],columns=list('ABC'))
print(test)

test2=pd.concat([test,multidata],axis=1)
print(test2)

A	B	C	(M1, W1)	(M1, W2)	(M1, W3)	(M1, W4)	(M1, W5)	(M2, W1)	(M2, W2)	(M2, W3)	(M2, W4)	(M2, W5)	(M3, W1)	(M3, W2)	(M3, W3)	(M3, W4)	(M3, W5)

0 1.5 0.5 -1.1 -0.8 0.9 0.6 -1.7 -0.1 -0.3 -1.0 -1.1 -3.5 1.6 0.9 -2.2 1.0 0.5 -0.4
1 0.7 -0.3 0.9 -0.6 -0.3 -0.9 0.0 1.8 0.9 1.5 -0.3 1.4 0.6 -0.4 -0.7 -1.3 -1.0 0.1
2 0.4 -0.8 -0.6 -1.2 0.4 0.2 0.9 -0.5 -0.9 -0.8 1.5 -0.4 -1.6 -0.5 -0.9 -0.4 1.4 0.2
3 1.9 1.5 -1.6 0.6 0.5 -1.4 0.0 -1.7 -0.5 1.9 0.3 -0.4 -0.3 -1.4 -0.3 0.2 1.0 1.3

Answer 1

追加するDFをあらかじめ同じ階層を持つMultiIndexカラムにしておけば結合できます。

import pandas as pd
import numpy as np
columns = pd.MultiIndex.from_product([['A', 'B'], ['C', 'D']], names=['month', 'week'])

data = np.round(np.random.randn(3, 4), 1)
multidata = pd.DataFrame(data, columns=columns)
print(multidata)

test = pd.DataFrame(data=np.round(np.random.randn(3, 2), 1), columns=['X', 'Y'])
print(test)

test2 = test.copy()
test2.columns = pd.MultiIndex.from_product([['Append'], test.columns], names=columns.names)
print(test2)

test3 = pd.concat([test2, multidata], axis=1)
print(test3)

month    A         B     
week     C    D    C    D
0      0.1 -0.0 -0.4  0.8
1      0.8  0.8  0.8 -0.1
2      1.3 -1.3 -2.5  0.1
     X    Y
0  1.0  1.0
1  0.0  0.8
2 -0.0 -1.1
month Append     
week       X    Y
0        1.0  1.0
1        0.0  0.8
2       -0.0 -1.1
month Append         A         B     
week       X    Y    C    D    C    D
0        1.0  1.0  0.1 -0.0 -0.4  0.8
1        0.0  0.8  0.8  0.8  0.8 -0.1
2       -0.0 -1.1  1.3 -1.3 -2.5  0.1

Comments

[Milly]

かぶりました、tiitoi さんと内容は同じです。

[omanko]

どうもありがとうございました！

Answer 2

結合前に test の columns を MultiIndex に変換した上で結合を行ってください。

test.columns = pd.MultiIndex.from_tuples([("M4", x) for x in test])

    M4          
     A    B    C
0 -0.3  0.0  1.2
1 -0.8 -1.1 -0.5
2  0.9  2.1 -0.7
3  1.4 -1.4  0.6

import pandas as pd
import numpy as np

columns = pd.MultiIndex.from_product(
    [["M1", "M2", "M3"], ["W1", "W2", "W3", "W4", "W5"]], names=["month", "week"]
)
data = np.round(np.random.randn(4, 15), 1)
multidata = pd.DataFrame(data, columns=columns)

test = pd.DataFrame(
    data=np.round(np.random.randn(4, 3), 1), index=[0, 1, 2, 3], columns=list("ABC")
)
test.columns = pd.MultiIndex.from_tuples([("M4", x) for x in test])

test2 = pd.concat([multidata, test], axis=1)

Comments

[omanko]

どうもありがとうございました！