如何使用 pandas python 绘制堆积条形图

Question

我有 3 个数据框用于年度数据（一个用于 2014 年、2015 年和 2016 年），每个数据框有 3 个名为 'PRACTICE', 'BNF NAME', 'ITEMS' 的列。

BNF NAME 是指药物名称，我正在挑选 3 个氨苄西林、阿莫西林和 Co-Amoxiclav。此列有不同的 strengths/dosages（例如 Co-Amoxiclav 200mg 或 Co-Amoxiclav 300mg 等）我想忽略，所以我使用了 str.contains() 到 select 这 3 种药物。 ITEMS是每种药物的总处方数。

我想创建一个堆叠条形图，其中 x 轴是年份（2014、2014、2015），y 轴是处方总数，3 个条形中的每一个都被分成 3 个药名.

我假设我可能需要使用 df.groupby() 和 select 部分字符串，但是我不确定如何合并年度数据以及如何对数据进行分组以创建堆积条图表。

任何指导将不胜感激。

这是我用来 select 仅显示 3 种药物名称的行的代码行。

frame=frame[frame['BNF NAME'].str.contains('Ampicillin' and 'Amoxicillin' and 'Co-Amoxiclav')]

这是每个数据帧的样子：

PRACTICE |  BNF NAME |  ITEMS
Y00327 | Co-Amoxiclav_Tab 250mg/125mg | 23
Y00327 | Co-Amoxiclav_Susp 125mg/31mg/5ml S/F | 10
Y00327 | Co-Amoxiclav_Susp 250mg/62mg/5ml S/F | 6
Y00327 | Co-Amoxiclav_Susp 250mg/62mg/5ml | 1
Y00327 | Co-Amoxiclav_Tab 500mg/125mg | 50

Answer 1

您可能会通过几种不同的方式来完成此操作。这是我会怎么做。我使用的是 jupyter 笔记本，所以你的 matplotlib 导入可能不同。

import pandas as pd
%matplotlib
import matplotlib.pyplot as plt
import matplotlib
matplotlib.style.use('ggplot')

df = pd.DataFrame({'PRACTICE': ['Y00327', 'Y00327', 'Y00327', 'Y00327', 'Y00327'],
                   'BNF NAME': ['Co-Amoxiclav_Tab 250mg/125mg', 'Co-Amoxiclav_Susp 125mg/31mg/5ml S/F',
                                'Co-Amoxiclav_Susp 250mg/62mg/5ml S/F', 'Ampicillin 250mg/62mg/5ml',
                               'Amoxicillin_Tab 500mg/125mg'],
                   'ITEMS': [23, 10, 6, 1, 50]})



Out[52]:
BNF NAME    ITEMS   PRACTICE
0   Co-Amoxiclav_Tab 250mg/125mg    23  Y00327
1   Co-Amoxiclav_Susp 125mg/31mg/5ml S/F    10  Y00327
2   Co-Amoxiclav_Susp 250mg/62mg/5ml S/F    6   Y00327
3   Ampicillin 250mg/62mg/5ml   1   Y00327
4   Amoxicillin_Tab 500mg/125mg 50  Y00327

模拟你的三个数据帧：

df1 = df.copy()
df2 = df.copy()
df3 = df.copy()

设置一列指示数据框代表的年份。

df1['YEAR'] = 2014
df2['YEAR'] = 2015
df3['YEAR'] = 2016

合并三个数据帧：

combined_df = pd.concat([df1, df2, df3], ignore_index=True)

设置每行代表什么药物：

combined_df['parsed_drug_name'] = "" # creates a blank column
amp_bool = combined_df['BNF NAME'].str.contains('Ampicillin', case=False)
combined_df.loc[amp_bool, 'parsed_drug_name'] = 'Ampicillin' # sets the row to amplicillin, if BNF NAME contains 'ampicillin.'

amox_bool = combined_df['BNF NAME'].str.contains('Amoxicillin', case=False)
combined_df.loc[amox_bool, 'parsed_drug_name'] = 'Amoxicillin'

co_amox_bool = combined_df['BNF NAME'].str.contains('Co-Amoxiclav', case=False)
combined_df.loc[co_amox_bool, 'parsed_drug_name'] = 'Co-Amoxiclav'

最后，对数据进行透视，并绘制结果：

combined_df.pivot_table(index='YEAR', columns='parsed_drug_name', values='ITEMS', aggfunc='sum').plot.bar(rot=0, stacked=True)

如何使用 pandas python 绘制堆积条形图

How to plot a stacked bar chart using pandas python

python

pandas

group-by

bar-chart

stacked