数据分析小实践：统计每个国家存在心理健康问题的平均年龄

# -*- coding:utf-8 -*-
"""
统计每个国家存在心理健康问题的平均年龄
"""
import csv
import matplotlib.pyplot as plt
import numpy as np
import matplotlib
data_depth = "./data/survey.csv"
dict = {}
result = {}
matplotlib.use('qt4agg')#指定默认字体  
matplotlib.rcParams['font.sans-serif']=['SimHei']
matplotlib.rcParams['font.family']='sans-serif'
with open(data_depth,'r',newline='') as data:
    rows = csv.reader(data)
    for i,row in enumerate(rows):
        if i == 0:
            continue
        country = row[3]
        age = row[1]
        if country not in dict:
            dict[country] = [0,0]
        dict[country][0] = dict[country][0] + int(row[1])
        dict[country][1] = dict[country][1] + 1

for key in dict:
    result[key] = dict[key][0] / dict[key][1]
sorted_result = sorted(result.items(),key=lambda asd:asd[1],reverse=True)
del sorted_result[0]
print(sorted_result)

x_data = []
for i in sorted_result:
    x_data.append(i[0])
y_data =[]
for i in sorted_result:
    y_data.append(i[1])
print(sorted_result.__len__())

fig = plt.figure()
ax = fig.add_subplot(1,1,1)
ax.bar(np.arange(10),y_data[:10],color='b',alpha = 0.5)
ax.set_xticks(np.arange(10)+0.4)
ax.set_xticklabels(x_data[:10])
ax.set_xlabel('国家')
ax.set_ylabel('平均年龄')
ax.set_title("排名前10的国家存在心理健康问题的科技工作者平均年龄")
plt.show()