嵌套字典到多索引数据框，其中字典键是列标签 [英] Nested dictionary to multiindex dataframe where dictionary keys are column labels

问题描述

假设我有一本看起来像这样的字典:

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]}}

我想要一个看起来像这样的数据框:

 A Ba b a b0 1 6 2 71 2 7 3 82 3 8 4 93 4 9 5 14 5 1 6 2

有没有方便的方法来做到这一点?如果我尝试:

在 [99] 中:数据框(字典)出[99]:甲乙[1, 2, 3, 4, 5] [2, 3, 4, 5, 6]b [6, 7, 8, 9, 1] [7, 8, 9, 1, 2]

我得到一个数据框，其中每个元素都是一个列表.我需要的是一个多索引，其中每个级别对应于嵌套字典中的键以及对应于列表中每个元素的行，如上所示.我想我可以使用一个非常粗略的解决方案，但我希望可能有一些更简单的方法.

解决方案

Pandas 想要 MultiIndex 值作为元组，而不是嵌套的字典.最简单的方法是在尝试将字典传递给 DataFrame 之前将其转换为正确的格式:

<预><代码>>>>改革 = {(outerKey，innerKey):outerKey 的值，innerDict 在 dictionary.iteritems() 中的innerKey，值在innerDict.iteritems()}>>>改革{('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(改革)甲乙a b a b0 1 6 2 71 2 7 3 82 3 8 4 93 4 9 5 14 5 1 6 2[5 行 x 4 列]

Say I have a dictionary that looks like this:

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]}}

and I want a dataframe that looks something like this:

     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

Is there a convenient way to do this? If I try:

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]

I get a dataframe where each element is a list. What I need is a multiindex where each level corresponds to the keys in the nested dict and the rows corresponding to each element in the list as shown above. I think I can work a very crude solution but I'm hoping there might be something a bit simpler.

解决方案

Pandas wants the MultiIndex values as tuples, not nested dicts. The simplest thing is to convert your dictionary to the right format before trying to pass it to 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]

这篇关于嵌套字典到多索引数据框，其中字典键是列标签的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！