Как сделать шкалу столбчатых диаграмм с панорамированными диаграммами до 100%? - PullRequest
Новая
       21

Как сделать шкалу столбчатых диаграмм с панорамированными диаграммами до 100%?

1 голос
/ 22 мая 2019

Я пытаюсь составить столбчатую диаграмму с накоплением на основе количества разных категорий (столбец «Класс» в моем фрейме данных).

Мои данные также сгруппированы по другой категории (столбец «СТРАТ»).

У меня есть графики фактических чисел, но я хочу получить их как пропорции к общему количеству 100%.то есть.так что все столбцы имеют полную высоту диаграммы, как вертикальная круговая диаграмма, для каждой категории 'STRAT'.

Примерно так: enter image description here

Ниже приведено то, что я попробовал, и результат.

Пример - небольшая часть большого набора данных.

import pandas as pd

df_test= pd.read_excel('df.xlsx')

df_test

From    To  Interval (m)    Class   STRAT   Total %S
308     309    1            PAF     CBC     4.15
309     310    1            PAF     CBC     3.76
320     321    1            UC      CBC     0.85
330     331    1            UC      CBC     0.698
342     343    1            NAF     LBB     0.259
376     377    1            NAF     LBB     0.395
412     413    1            UC      LBB     1.19
51      52     1            PAF     UBB     2.27
420     420.5  0.5          UC      UAB     2.85
189     190    1            PAF     LBB     1.52
520     521    1            NAF     UAB     1.45
632     633    1            NAF     UAB     0.0615
644     645    1            NAF     UAB     0.178


df_test.groupby(['STRAT', 'Class']).size().unstack().plot.bar(stacked=True)

Это дает мне столбчатую диаграмму с накоплением, но она не масштабируется до 100%

stacked bar chart

Я не могу найтиаккуратный способ поместить вычисление "Class count / total_counts" в код, чтобы получить процент вместо числа.

1 Ответ

2 голосов
/ 22 мая 2019

Есть много вещей, которые вы можете сделать с помощью matplotlib, чтобы принудительно масштабировать ось y, чтобы она нормализовала все до 100%, как показано здесь: 100% столбчатая диаграмма в MatPlotLib

ОднакоВаша проблема может быть решена намного проще.

Если это ваш фрейм данных с двумя ключевыми столбцами, Class и STRAT: 

import pandas as pd
import numpy as np
df = pd.DataFrame({'Class': 2*['a', 'b', 'c', 'd'],'STRAT': 'x', 'y', 'x', 'z','y','x','z','w'],'value': np.random.randint(0, int(1e2),8)})

Тогда вы можете легко рассчитать% ges следующим образом: 

df_pct = (df.groupby(['STRAT','Class'])['value'].count()/df.groupby(['STRAT'])['value'].count())

Инаконец, вы можете построить график точно так же, как в своем коде: 

df_pct.unstack().plot.bar(stacked=True)
...