matplotlib画基础图形

最新推荐文章于 2024-10-23 17:59:48 发布

hzq201435

最新推荐文章于 2024-10-23 17:59:48 发布

阅读量147

点赞数

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/hzq201435/article/details/124210904

版权

python 专栏收录该内容

10 篇文章 0 订阅

订阅专栏

这段代码使用Python的matplotlib库创建了一幅柱状图，展示了一个数据集中关于心脏病（HeartDisease）与不同年龄类别（AgeCategory）的关系。它首先设置图像大小，然后统计每个年龄类别中患心脏病的人数，并绘制了相应的柱状图。最后，保存了图像并显示出来，帮助分析心脏病在各年龄段的分布情况。

摘要由CSDN通过智能技术生成

import numpy as np
import matplotlib.pyplot as plt
fig,ax = plt.subplots()
# 设置图像大小
fig.set_size_inches(12,6)
# plt.rcParams('figure.figsize')=(12,8)
# 统计不同数据
n = len(df)
res_num = len(e_sort)
print(list(e_sort.keys()))
key_num = dict()
# 初始化为0
for k,v in e_sort.items():
    key_num[k] = 0
for i in range(n):
    for k in e_sort.keys():
        if df['HeartDisease'][i]=='Yes' and df['AgeCategory'][i] ==k:
            key_num[k]+=1
ax.set_xlabel('AgeCategory',fontsize=12) #设置x轴，y轴的标题
ax.set_ylabel('HeartDisease Num',fontsize=12)
y1 = np.array(list(key_num.values()))
# x1 = np.array([0,1])
x1 = np.arange(res_num)
# 柱形的宽度
bar_width = 0.5
# 绘制柱形图
plt.bar(x1, y1, tick_label=list(e_sort.keys()), width=bar_width)
# 保存
plt.savefig('AgeCategory')
plt.show()