python - recorrer - Diccionario anidado a marco de datos de múltiples índices donde las claves del diccionario son etiquetas de columnas

lista de diccionarios python

Question

Digamos que tengo un diccionario que se ve así:

dictionary = {''A'' : {''a'': [1,2,3,4,5],
                     ''b'': [6,7,8,9,1]},

              ''B'' : {''a'': [2,3,4,5,6],
                     ''b'': [7,8,9,1,2]}}

y quiero un marco de datos que se parece a esto:

     A   B
     a b a b
  0  1 6 2 7
  1  2 7 3 8
  2  3 8 4 9
  3  4 9 5 1
  4  5 1 6 2

¿Hay una manera conveniente de hacer esto? Si intento:

In [99]:

DataFrame(dictionary)

Out[99]:
     A               B
a   [1, 2, 3, 4, 5] [2, 3, 4, 5, 6]
b   [6, 7, 8, 9, 1] [7, 8, 9, 1, 2]

Obtengo un marco de datos donde cada elemento es una lista. Lo que necesito es un multi índice donde cada nivel corresponda a las claves en el dictado anidado y las filas correspondientes a cada elemento en la lista como se muestra arriba. Creo que puedo trabajar en una solución muy cruda, pero espero que haya algo un poco más simple.

Answer 1

Pandas quiere los valores de MultiIndex como tuplas, no como dictados anidados. Lo más simple es convertir el diccionario al formato correcto antes de intentar pasarlo a DataFrame:

>>> reform = {(outerKey, innerKey): values for outerKey, innerDict in dictionary.iteritems() for innerKey, values in innerDict.iteritems()}
>>> reform
{(''A'', ''a''): [1, 2, 3, 4, 5],
 (''A'', ''b''): [6, 7, 8, 9, 1],
 (''B'', ''a''): [2, 3, 4, 5, 6],
 (''B'', ''b''): [7, 8, 9, 1, 2]}
>>> pandas.DataFrame(reform)
   A     B   
   a  b  a  b
0  1  6  2  7
1  2  7  3  8
2  3  8  4  9
3  4  9  5  1
4  5  1  6  2

[5 rows x 4 columns]

Answer 2

dict_of_df = {k: pd.DataFrame(v) for k,v in dictionary.items()}
df = pd.concat(dict_of_df, axis=1)

Tenga en cuenta que el orden de las columnas se pierde para python <3.6