Как сделать несколько гистограмм / гистограмм, связанных с двумя столбцами в python? - PullRequest
Новая
       23

Как сделать несколько гистограмм / гистограмм, связанных с двумя столбцами в python?

0 голосов
/ 23 января 2020

У меня есть набор данных с несколькими строками и столбцами. Упрощение: 

A   B   

g   1    
h   2   
h   3   
g   3   
j   4   
g   5

Где A - код операции, а B - продолжительность пребывания в больнице. Итак, каждая из строк соответствует пациенту.

Я хотел бы сделать несколько графиков (по одному для каждой категории) столбцов A и B. В этом примере:

График 1: Один график, на котором «х» представляет каждого пациента, которому была сделана операция «g», а «y» представляет продолжительность пребывания каждого пациента. Таким образом, для этого графика должно существовать 3 столбца, так как есть 3 пациента с операцией "g";

График 2: Один график, где x представляет каждого пациента, который прошел операцию "h", а y представляет продолжительность пребывания каждого пациента. Таким образом, для этого графика должны существовать 2 столбца, так как есть 2 пациента с операцией «h»;

График 3: Один график, где x представляет каждого пациента, который прошел операцию «j», а y представляет продолжительность пребывания каждого пациента. Таким образом, для этого графика должно существовать 1 столбец, так как 1 пациент с операцией "j";

Кто-нибудь может мне помочь?

1 Ответ

1 голос
/ 23 января 2020

Вы можете сделать это, получить набор данных в pandas как df: 

>>> import pandas as pd
>>> import matplotlib.pyplot as plt
>>> df = pd.DataFrame({'A':['g', 'h', 'h', 'g', 'j', 'g'], 'B':[1, 2, 3, 3, 4, 5]})
>>> df
   A  B
0  g  1
1  h  2
2  h  3
3  g  3
4  j  4
5  g  5
>>> df.groupby('A').plot(kind = 'bar')
A
g    AxesSubplot(0.125,0.11;0.775x0.77)
h    AxesSubplot(0.125,0.11;0.775x0.77)
j    AxesSubplot(0.125,0.11;0.775x0.77)
dtype: object
>>> plt.show()   #you'll see 3 different figures for that.
...