抑郁症患者数据分析
导入数据
import pandas as pd
from pyecharts.charts import *
from pyecharts import options as optsdf=pd.read_csv('YiYuZheng.csv')
df.head(1)
| Patient_name | Label | Date | Title | Communications | Doctor | Hospital | Faculty | |
|---|---|---|---|---|---|---|---|---|
| 0 | 患者：女 43岁 | 压抑 | 05.28 | 压抑 个人情况：去年1月份开始夫妻两地分居，孩子13岁男孩住校，平... 这种情况是否需要去... | 115 | 杨胜文 | 襄阳市安定医院 | 心理科 |
# 查看数据
df.info()
<class 'pandas.core.frame.DataFrame'>
RangeIndex: 8400 entries, 0 to 8399
Data columns (total 8 columns):# Column Non-Null Count Dtype
--- ------ -------------- ----- 0 Patient_name 8400 non-null object1 Label 8400 non-null object2 Date 8288 non-null object3 Title 8400 non-null object4 Communications 8400 non-null int64 5 Doctor 8400 non-null object6 Hospital 8400 non-null object7 Faculty 8400 non-null object
dtypes: int64(1), object(7)
memory usage: 525.1+ KB
从数据反馈结果来看：Date列存在空缺值，并且不是日期类型。
Patient_name列存在信息混合一起情况，需要拆分年龄和性别。
数据预处理
拆分年龄和性别
#获取性别，作为新列
#患者：女 43岁,首先按照空格拆分，结果为[患者：女]\[ ]\[43岁],选取第一个，第二次按照中文冒号拆分，[患者][: ][女]
df['Sex']=df['Patient_name'].map(lambda x:x.split(" ")[0]).map(lambda x:x.split("：")[-1])#获取年龄，作为新列
#患者：女 43岁,首先按照空格拆分，结果为[患者：女]\[ ]\[43岁],选取第三个，并且去掉“岁”
df['Age']=df['Patient_name'].map(lambda x:x.split(" ")[2][:-1])df.head()
| Patient_name | Label | Date | Title | Communications | Doctor | Hospital | Faculty | Sex | Age | |
|---|---|---|---|---|---|---|---|---|---|---|
| 0 | 患者：女 43岁 | 压抑 | 05.28 | 压抑 个人情况：去年1月份开始夫妻两地分居，孩子13岁男孩住校，平... 这种情况是否需要去... | 115 | 杨胜文 | 襄阳市安定医院 | 心理科 | 女 | 43 |
| 1 | 患者：女 32岁 | 生气。心梗。抑郁 | 05.28 | 生气。心梗。抑郁 郁郁寡欢。被他人语言刺激。卧床不起。没动力。心疼。受伤 是什么病。怎么办 | 12 | 郭汉法 | 泰安八十八医院 | 临床心理科 | 女 | 32 |
| 2 | 患者：女 15岁 | 情绪低落，烦躁抑郁 | 05.28 | 情绪低落，烦躁抑郁 情绪低落，压抑烦躁，思考能力降低。长时间学习，睡眠时间少。睡... 还有... | 2 | 郭苏皖 | 南京脑科医院 | 医学心理科 | 女 | 15 |
| 3 | 患者：女 16岁 | 抑郁 | 05.28 | 抑郁 前面已简述，2024年夏季中考，本来学习非常好，非常自律，自... 已经服用9个月的艾... | 2 | 刘丽 | 联勤保障部队第九〇四医院（常州院区） | 精神3科（物质依赖科） | 女 | 16 |
| 4 | 患者：女 67岁 | 焦虑症 严重躯干反应、抑郁症 | 05.28 | 焦虑症 严重躯干反应 抑郁症 草酸加量以后，还是有比较严重的躯干反应，主要表现为背痛 脖... | 2 | 刘晓华 | 上海市精神卫生中心 | 精神科 | 女 | 67 |
处理空缺值
df.isnull().sum()
Patient_name 0
Label 0
Date 112
Title 0
Communications 0
Doctor 0
Hospital 0
Faculty 0
Sex 0
Age 0
dtype: int64
#因为空缺数据较少，并且不适合使用填充法，故而删除
df.dropna(inplace=True)#在原来的数据上删除
df.info()
<class 'pandas.core.frame.DataFrame'>
Int64Index: 8288 entries, 0 to 8399
Data columns (total 10 columns):# Column Non-Null Count Dtype
--- ------ -------------- ----- 0 Patient_name 8288 non-null object1 Label 8288 non-null object2 Date 8288 non-null object3 Title 8288 non-null object4 Communications 8288 non-null int64 5 Doctor 8288 non-null object6 Hospital 8288 non-null object7 Faculty 8288 non-null object8 Sex 8288 non-null object9 Age 8288 non-null object
dtypes: int64(1), object(9)
memory usage: 712.2+ KB
修改Date列
#df['Date']
#转换成字符串类型
df['Date']=df['Date'].astype(str)#定义函数，实现date列格式统一：年-月-日
def trans_date(tag):if tag.startswith("20"):#查看是否以20开头，即查看是否存在年tag=tag.replace(".","-")else:tag="2025-"+tag.replace(".","-")#否则加上年份return tagdf['Date']= df['Date'].map(lambda x:trans_date(x))#调用函数转换格式#转换成日期类型
df['Date']=pd.to_datetime(df['Date'])#df.info()
数据可视化分析
from pyecharts.globals import ThemeType #导入主题库
查看患者性别分布情况
#准备数据：按照性别统计个数
data=df['Sex'].value_counts()
#data
x=data.index.tolist()
y=data.tolist()#绘制饼图
pie=(Pie(init_opts=opts.InitOpts(theme=ThemeType.LIGHT))#设置主题.add("",[list(z) for z in zip(x,y)],#数据需要打包成[(key,value),(key,value),...]label_opts=opts.LabelOpts(formatter="{b}:{d}%")#以百分比形式显示标签).set_global_opts(title_opts=opts.TitleOpts(title="患者性别分布情况"))
)
pie.render_notebook()
可见：女性抑郁症更为常见。
患者年龄分布情况
#数据准备
#1.转换年龄为数值类型
#df['Age']=df['Age'].astype(int)
#因为年龄数据不规范，存在：X岁Y月 形式的数据，再次进行数据处理
df['Age']=df['Age'].map(lambda x:"1" if ("天" in x or "个" in x or "月" in x) else x).astype(int)
#df.info()
#2.年龄分箱
labels=["0~18","19~40","41~60","61~80","80+"]#区间标签
df['age_label']=pd.cut(df['Age'],bins=[0,18,40,60,80,100],labels=labels)#分箱#3.统计各个年龄区间人数
data=df['age_label'].value_counts().sort_index()#按照索引值排序
#datax=data.index.tolist()
y=data.tolist()#画柱状图
bar=(Bar(init_opts=opts.InitOpts(theme=ThemeType.DARK))#主题配置.add_xaxis(x).add_yaxis("人数",y).set_global_opts(title_opts=opts.TitleOpts(title="患者年龄分布情况")).reversal_axis()
)
bar.render_notebook()
查看患者就诊次数分布
#对就诊次数分箱
bins=[0,5,10,20,40,100,200,300]
df['Communications_count']=pd.cut(df['Communications'],bins=bins)
#按照就诊分箱数据统计数据
data=df['Communications_count'].value_counts().sort_index()#按照索引排序
#data
x=data.index.astype(str).tolist()
y=data.tolist()#画柱状图
line=(Line(init_opts=opts.InitOpts(theme=ThemeType.DARK))#主题配置.add_xaxis(x).add_yaxis("人数",y).set_global_opts(title_opts=opts.TitleOpts(title="患者就诊分布情况"),xaxis_opts=opts.AxisOpts(type_='category'),#x轴数据为类别类型yaxis_opts=opts.AxisOpts(type_='value'),)#y轴数据为数值类型)
line.render_notebook()
<div id="22f20d64280d408c96ed5b60aa15391a" style="width:900px; height:500px;"></div>
患者标签分布
#pip install jieba
^C
Note: you may need to restart the kernel to use updated packages.
import jieba
from collections import Counter
#定义函数分词
def chinese_word_cut(text):seg_list=jieba.cut(text,cut_all=False)return [word for word in seg_list if len(word)>1]#对表格数据进行分词
all_word=[]
for text in df['Title']:all_word.extend(chinese_word_cut(text))#统计词频
word_count=Counter(all_word)
top_words=word_count.most_common(100)#取前100个高频词data=[(word,count) for word,count in top_words]w=(WordCloud().add("",data)
)
w.render_notebook()
患者就诊的科室分布
data=df['Faculty'].value_counts()[:10] #选取前十科室pie=(Pie().add('',[list(z) for z in zip(data.index.tolist(),data.tolist())],#饼图数据格式[[key1,value1],[key2,value2],...]label_opts=opts.LabelOpts(formatter="{b}:{d}%")#标签格式)
)
pie.render_notebook()
