以下是完全来自AI的脚本,给到的指令是:
python 脚本,柱状图,我有一个多年份的统计表,输入文件第一列为横坐标“药物”,输入文件第二列及之后的列为每一年的数据,每一行的数据为一种药物对应的多年份的耐药数据,每个单元格对应的是该药物的耐药个数,及占总数的比列,百分比列写在了中文的括号内。请画柱状图,每种药物的多年份的耐药个数及比例,我的输入文件是“test.xlsx”
输入示例:
药物 | 2022 | 2023 | 2024 |
---|---|---|---|
药物1 | 3 (25%) | 4 (55%) | 3 (25%) |
药物2 | 3 (25%) | 4 (35%) | 2 (15%) |
import pandas as pd
import matplotlib.pyplot as plt
import matplotlib.ticker as ticker
# 设置字体为中文字体
plt.rcParams['font.sans-serif'] = ['SimSun']
# 读取Excel文件
df = pd.read_excel('test.xlsx')
# 提取药物和年份列
drugs = df.iloc[:, 0].tolist()
years = df.columns[1:].tolist()
# 提取每种药物的多年份耐药个数和比例数据
data = df.iloc[:, 1:].values.tolist()
# 绘制柱状图
fig, ax = plt.subplots()
width = 0.2
x = range(len(drugs))
for i in range(len(years)):
values = [row[i] for row in data]
percentages = [float(value.split('(')[1].split(')')[0]) for value in values]
bars = ax.bar([xi + width*i for xi in x], percentages, width, label=years[i])
# 在每个柱状条上添加百分比数字
for j, bar in enumerate(bars):
height = bar.get_height()
if height >= 5:
ax.text(bar.get_x() + bar.get_width() / 2, height, f'{height:.2f}%', ha='center', va='bottom', rotation=90, fontsize=6)
else:
ax.text(bar.get_x() + bar.get_width() / 2, height + 0.5, f'{height:.2f}%', ha='center', va='bottom', rotation=90, fontsize=6)
ax.margins(y=0.15)
plt.subplots_adjust(bottom=0.2)
# 设置图表标题和轴标签
ax.set_title('图1 XX菌耐药情况', fontsize=16)
ax.set_xlabel('药物', fontsize=14)
ax.set_ylabel('耐药比例', fontsize=14)
# 设置横坐标刻度和标签,并旋转横坐标标签
ax.yaxis.set_major_formatter(ticker.FormatStrFormatter('%.2f'))
ax.set_xticks([xi + width*(len(years)-1)/2 for xi in x])
ax.set_xticklabels(drugs, rotation=90)
# 添加图例,并将图例向右偏移
ax.legend(bbox_to_anchor=(1.02, 1), loc='upper left')
# 调整每种药物之间的间隔
plt.subplots_adjust(wspace=0.3)
#plt.show()
# 保存为PNG图片并设置足够宽的dpi
plt.savefig('results.png', dpi=800, bbox_inches='tight')
# 图片保存路径和文件名