Oi Eu tenho pesquisado os métodos concat, join e merge para pandas e parece que não consigo encontrar o que quero.
Vamos supor que eu tenho dois quadros de dados
A = pd.DataFrame("A",index=[0,1,2,3,4],columns=['Col 1','Col 2','Col 3'])
B = pd.DataFrame("B",index=[0,1,2,3,4],columns=['Col 1','Col 2','Col 3'])
>>> A
Col 1 Col 2 Col 3
0 A A A
1 A A A
2 A A A
3 A A A
4 A A A
>>> B
Col 1 Col 2 Col 3
0 B B B
1 B B B
2 B B B
3 B B B
4 B B B
Agora, quero criar um novo quadro de dados com as colunas mescladas, acho mais fácil explicar se eu criar um índice múltiplo de como quero as colunas
index = pd.MultiIndex.from_product([A.columns.values,['A','B']])
>>> index
MultiIndex(levels=[['Col 1', 'Col 2', 'Col 3'], ['A', 'B']],
labels=[[0, 0, 1, 1, 2, 2], [0, 1, 0, 1, 0, 1]])
Agora, se eu criar um quadro de dados vazio com esse índice múltiplo para as colunas
empty_df = pd.DataFrame('-',index=A.index,columns=index)
>>> empty_df
Col 1 Col 2 Col 3
A B A B A B
0 - - - - - -
1 - - - - - -
2 - - - - - -
3 - - - - - -
4 - - - - - -
Minha pergunta é: qual mesclagem, concat ou associação eu uso para obter isso? Eu tentei várias coisas para concat ... interior, exterior, etc. Não consigo encontrar o que quero. A única coisa em que consigo pensar é criar o quadro de dados vazio e depois preencher novamente.
Edit: Depois de experimentar a resposta de Jezrael, está perto, mas não exatamente. O que eu quero é como colunas aninhadas de classificação? Por exemplo
empty_df['Col 1']
>>> empty_df['Col 1']
A B
0 - -
1 - -
2 - -
3 - -
4 - -
Ou
>>> empty_df['Col 1']['A']
0 -
1 -
2 -
3 -
4 -
Name: A, dtype: object
Portanto, esta é uma solução que eu criei, mas é iterada nas colunas.
row_idx = A.index.union(B.index)
col_idx = pd.MultiIndex.from_product([A.columns.values,['A','B']])
new_df = pd.DataFrame('-',index=row_idx,columns=col_idx)
for column in A.columns:
new_df.loc[:,(column,'A')] = A[column]
new_df.loc[:,(column,'B')] = B[column]
>>> new_df
Col 1 Col 2 Col 3
A B A B A B
0 A B A B A B
1 A B A B A B
2 A B A B A B
3 A B A B A B
4 A B A B A B
>>> new_df['Col 1']
A B
0 A B
1 A B
2 A B
3 A B
4 A B
>>> new_df['Col 1']['A']
0 A
1 A
2 A
3 A
4 A
Name: A, dtype: object