Объедините несколько словарей в один кадр данных панд в длинном формате - PullRequest
0 голосов
/ 03 декабря 2018

У меня есть несколько словарей, настроенных следующим образом:

Dict1 = {'Orange': ['1', '2', '3', '4']}
Dict2 = {'Red': ['3', '4', '5']}

И я хотел бы, чтобы вывод был одним объединенным фреймом данных:

| Type | Value |
|--------------|
|Orange|   1   |
|Orange|   2   |
|Orange|   3   |
|Orange|   4   |
| Red  |   3   |
| Red  |   4   |
| Red  |   5   |

Я попытался разделить все, но яполучите только Dict2 на этом фрейме данных.

mydicts = [Dict1, Dict2]
for x in mydicts:
    for k, v in x.items():
        df = pd.DataFrame(v)
        df['Type'] = k

Ответы [ 3 ]

0 голосов
/ 04 декабря 2018

Понимание

pd.DataFrame(
    [(t, v)
     for t, V in {**Dict1, **Dict2}.items()
     for v in V],
    columns=['Type', 'Value']
)

     Type Value
0  Orange     1
1  Orange     2
2  Orange     3
3  Orange     4
4     Red     3
5     Red     4
6     Red     5
0 голосов
/ 04 декабря 2018

После stack это становится unnest проблемой

s=pd.DataFrame(mydicts).stack().reset_index(level=1)
unnesting(s,[0])
Out[829]: 
   0 level_1
0  1  Orange
0  2  Orange
0  3  Orange
0  4  Orange
1  3     Red
1  4     Red
1  5     Red
0 голосов
/ 03 декабря 2018

Один вариант использует pd.concat:

pd.concat(map(pd.DataFrame, mydicts), axis=1).melt().dropna()

  variable value
0   Orange     1
1   Orange     2
2   Orange     3
3   Orange     4
4      Red     3
5      Red     4
6      Red     5

Если производительность имеет значение, вы можете инициализировать один DataFrame, используя DataFrame.from_dict и melt:

pd.DataFrame.from_dict({**Dict1, **Dict2}, orient='index').T.melt().dropna()
  variable value
0   Orange     1
1   Orange     2
2   Orange     3
3   Orange     4
4      Red     3
5      Red     4
6      Red     5

Или, используя stack вместо melt (немного медленнее, просто для полноты):

res = (pd.DataFrame.from_dict({**Dict1, **Dict2}, orient='index').T
         .stack()
         .reset_index(level=1)
         .sort_values('level_1'))
res.columns = ['Type', 'Value']

print(res)
     Type Value
0  Orange     1
1  Orange     2
2  Orange     3
3  Orange     4
0     Red     3
1     Red     4
2     Red     5

Синтаксис распаковки словаря работает с python3.6.В старых версиях замените {**d1, **d2} на {k: v for d in mydicts for k, v in d.items()}.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...