Как мне построить график пропорций в Python, используя .groupby - PullRequest
0 голосов
/ 04 марта 2020

Я пытаюсь создать гистограмму пропорций в python. В моем наборе данных есть два столбца для двух разных групп («Timing_Cond» и «Test_Cond», каждая из которых содержит 2 значения, представляющих по два условия в каждом) и столбец («Test_Model_Opt»), представляющий, подходит ли каждая строка наилучшим образом для указанной модели c ( 0 за это не было и 1 за это подгонял модель).

Это мой код для отображения количества каждой группы для тех, которые соответствуют модели, и тех, которые не соответствуют модели. Вместо этого я хочу отобразить процент соответствия каждой группы.

Так, например, у меня есть группа A из Timing_Cond и группа A из Test_Cond в качестве одного условия. Я хочу построить график для этой GroupA / GroupA Процент тех, которые подходят для модели A (есть только две модели, опять же представленные 0 или 1). Что было бы лучшим способом go сделать это?

dfall['Test_Model_Opt'] = dfall['Test_Model_Opt'].astype('category')
fig, ax = plt.subplots(figsize=(15,7))
dfall.groupby(['Test_Model_Opt','Timing_Cond', 'Test_Cond'])['Test_Model_Opt'].count().unstack(0).plot.bar(ax=ax)
ax.set_xlabel('Condition', fontsize=20)
ax.set_ylabel('Count', fontsize=20)
plt.title('Model Counts', fontsize=20)
plt.ylim(0,1)
plt.xticks([0,1,2,3] , ['Immediate, Study','Immediate, Test', 'Delay, Study', 'Delay, Test'], rotation=0)
plt.rc('xtick', labelsize=20) 
plt.rc('ytick', labelsize=15) 
ax.yaxis.grid(True)
ax.legend(('Nondeclarative Strategy','Declarative Strategy'), fontsize=20)
plt.savefig('C:/Users/Pablo/Desktop/WPT Model Fits/plots/allcounts.png')
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...