如何在python中制作與兩列相關的多個直方圖/條形圖？

Question

我有一個包含多行和多列的數據集。 簡化：

A   B   

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

其中 A 是手術代碼，B 是住院時間。 因此，每一行對應一個患者。

我想做 A 列和 B 列的多個圖（每個類別一個）。在這個例子中：

圖 1：一個圖，其中 x 代表每位接受“g”手術的患者，y 代表每位患者的住院時間。 所以對於這個情節，必須存在 3 個條形，因為有 3 個患者進行了“g”手術；

圖 2：一個圖，其中 x 代表每位接受“h”手術的患者，y 代表每位患者的住院時間。 所以對於這個情節，必須存在 2 個條形，因為有 2 個患者接受了“h”手術；

圖 3：一個圖，其中 x 代表每位接受“j”次手術的患者，y 代表每位患者的住院時間。 所以對於這個情節，必須存在 1 個條形，因為有 1 個患者接受了“j”手術；

誰能幫我？

Answer 1

您可以這樣做，將 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.