'commit'
This commit is contained in:
252
Test/get_education_data_by_year.py
Normal file
252
Test/get_education_data_by_year.py
Normal file
@@ -0,0 +1,252 @@
|
||||
import json
|
||||
import os
|
||||
|
||||
# 定义学段映射关系
|
||||
EDUCATION_STAGES = {
|
||||
'preschool': '学前',
|
||||
'primary': '小学',
|
||||
'junior': '初中',
|
||||
'senior': '高中',
|
||||
'vocational': '中职'
|
||||
}
|
||||
|
||||
# 数据文件路径
|
||||
DATA_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), "../Data"))
|
||||
SCHOOL_COUNT_FILE = os.path.join(DATA_DIR, "SchoolCount.json")
|
||||
TEACHER_COUNT_FILE = os.path.join(DATA_DIR, "TeacherCount.json")
|
||||
ENROLLMENT_COUNT_FILE = os.path.join(DATA_DIR, "ZaiXiaoShengCount.json")
|
||||
ADMISSION_COUNT_FILE = os.path.join(DATA_DIR, "ZhaoShengCount.json")
|
||||
|
||||
|
||||
def load_json_file(file_path):
|
||||
"""加载JSON文件数据"""
|
||||
try:
|
||||
with open(file_path, "r", encoding="utf-8") as f:
|
||||
return json.load(f)
|
||||
except Exception as e:
|
||||
print(f"加载文件 {file_path} 失败: {e}")
|
||||
return []
|
||||
|
||||
|
||||
def get_province_data(data_list):
|
||||
"""从数据列表中获取云南省的数据"""
|
||||
if not isinstance(data_list, list):
|
||||
print(f"数据列表不是预期的列表类型: {type(data_list)}")
|
||||
return None
|
||||
|
||||
for item in data_list:
|
||||
if isinstance(item, dict) and item.get("area_name") == "云南省":
|
||||
return item
|
||||
return None
|
||||
|
||||
|
||||
def get_school_count_by_year_and_stage(year, stage):
|
||||
"""获取指定年份和学段的学校总数"""
|
||||
school_data = load_json_file(SCHOOL_COUNT_FILE)
|
||||
province_data = get_province_data(school_data)
|
||||
|
||||
if not province_data:
|
||||
return 0
|
||||
|
||||
# 根据学段获取对应的学校数据字段名
|
||||
stage_field_mapping = {
|
||||
'preschool': 'preschool_schools',
|
||||
'primary': 'primary_schools',
|
||||
'junior': 'junior_schools',
|
||||
'senior': 'senior_schools',
|
||||
'vocational': 'vocational_schools'
|
||||
}
|
||||
|
||||
field_name = stage_field_mapping.get(stage)
|
||||
if not field_name or field_name not in province_data:
|
||||
print(f"学段 {stage} 对应的学校数据字段不存在")
|
||||
return 0
|
||||
|
||||
year_data = province_data[field_name].get(str(year))
|
||||
if not year_data or not isinstance(year_data, dict):
|
||||
print(f"年份 {year} 的学校数据不存在或格式不正确")
|
||||
return 0
|
||||
|
||||
# 计算城区、镇区、乡村的学校总数
|
||||
try:
|
||||
total_schools = year_data.get('urban', 0) + year_data.get('town', 0) + year_data.get('rural', 0)
|
||||
return total_schools
|
||||
except Exception as e:
|
||||
print(f"计算学校总数时出错: {e}")
|
||||
return 0
|
||||
|
||||
|
||||
def get_teacher_count_by_year_and_stage(year, stage):
|
||||
"""获取指定年份和学段的教职工总数"""
|
||||
teacher_data = load_json_file(TEACHER_COUNT_FILE)
|
||||
province_data = get_province_data(teacher_data)
|
||||
|
||||
if not province_data:
|
||||
return 0
|
||||
|
||||
# 根据学段获取对应的教职工数据字段名
|
||||
stage_field_mapping = {
|
||||
'preschool': 'preschool_teachers',
|
||||
'primary': 'primary_teachers',
|
||||
'junior': 'junior_teachers',
|
||||
'senior': 'senior_teachers',
|
||||
'vocational': 'vocational_teachers'
|
||||
}
|
||||
|
||||
field_name = stage_field_mapping.get(stage)
|
||||
if not field_name or field_name not in province_data:
|
||||
print(f"学段 {stage} 对应的教职工数据字段不存在")
|
||||
return 0
|
||||
|
||||
year_data = province_data[field_name].get(str(year))
|
||||
if not year_data or not isinstance(year_data, dict):
|
||||
print(f"年份 {year} 的教职工数据不存在或格式不正确")
|
||||
return 0
|
||||
|
||||
# 优先使用total_staff，如果为0或不存在，则返回教师总数
|
||||
try:
|
||||
total_staff = year_data.get('total_staff', 0)
|
||||
if total_staff == 0:
|
||||
total_staff = year_data.get('total_teacher', 0)
|
||||
return total_staff
|
||||
except Exception as e:
|
||||
print(f"获取教职工总数时出错: {e}")
|
||||
return 0
|
||||
|
||||
|
||||
def get_enrollment_count_by_year_and_stage(year, stage):
|
||||
"""获取指定年份和学段的在校生总数"""
|
||||
enrollment_data = load_json_file(ENROLLMENT_COUNT_FILE)
|
||||
province_data = get_province_data(enrollment_data)
|
||||
|
||||
if not province_data or 'student_data' not in province_data:
|
||||
print("未找到学生数据")
|
||||
return 0
|
||||
|
||||
stage_data = province_data['student_data'].get(stage)
|
||||
if not stage_data:
|
||||
print(f"学段 {stage} 的学生数据不存在")
|
||||
return 0
|
||||
|
||||
year_data = stage_data.get(str(year))
|
||||
if not year_data or not isinstance(year_data, dict):
|
||||
print(f"年份 {year} 的学生数据不存在或格式不正确")
|
||||
return 0
|
||||
|
||||
# 返回在校生总数
|
||||
try:
|
||||
return year_data.get('total', 0)
|
||||
except Exception as e:
|
||||
print(f"获取在校生总数时出错: {e}")
|
||||
return 0
|
||||
|
||||
|
||||
def get_admission_count_by_year_and_stage(year, stage):
|
||||
"""获取指定年份和学段的招生总数"""
|
||||
admission_data = load_json_file(ADMISSION_COUNT_FILE)
|
||||
province_data = get_province_data(admission_data)
|
||||
|
||||
if not province_data or 'education_data' not in province_data:
|
||||
print("未找到教育数据")
|
||||
return 0
|
||||
|
||||
stage_data = province_data['education_data'].get(stage)
|
||||
if not stage_data:
|
||||
print(f"学段 {stage} 的教育数据不存在")
|
||||
return 0
|
||||
|
||||
year_data = stage_data.get(str(year))
|
||||
if not year_data or not isinstance(year_data, dict):
|
||||
print(f"年份 {year} 的教育数据不存在或格式不正确")
|
||||
return 0
|
||||
|
||||
# 返回招生总数
|
||||
try:
|
||||
return year_data.get('total', 0)
|
||||
except Exception as e:
|
||||
print(f"获取招生总数时出错: {e}")
|
||||
return 0
|
||||
|
||||
|
||||
def get_education_data_by_year(year):
|
||||
"""获取指定年份所有学段的教育数据"""
|
||||
result = []
|
||||
|
||||
for stage_code, stage_name in EDUCATION_STAGES.items():
|
||||
# 获取各类数据（增加异常处理）
|
||||
try:
|
||||
school_count = get_school_count_by_year_and_stage(year, stage_code)
|
||||
teacher_count = get_teacher_count_by_year_and_stage(year, stage_code)
|
||||
enrollment_count = get_enrollment_count_by_year_and_stage(year, stage_code)
|
||||
admission_count = get_admission_count_by_year_and_stage(year, stage_code)
|
||||
|
||||
# 转换为万人单位
|
||||
teacher_count_10k = round(teacher_count / 10000, 2) if teacher_count else 0
|
||||
enrollment_count_10k = round(enrollment_count / 10000, 2) if enrollment_count else 0
|
||||
admission_count_10k = round(admission_count / 10000, 2) if admission_count else 0
|
||||
|
||||
# 添加到结果列表
|
||||
result.append({
|
||||
'education_stage': stage_name,
|
||||
'school_count': school_count,
|
||||
'teacher_count_10k': teacher_count_10k,
|
||||
'enrollment_count_10k': enrollment_count_10k,
|
||||
'admission_count_10k': admission_count_10k
|
||||
})
|
||||
except Exception as e:
|
||||
print(f"处理学段 {stage_name} 时出错: {e}")
|
||||
# 出错时添加一个默认的空数据条目
|
||||
result.append({
|
||||
'education_stage': stage_name,
|
||||
'school_count': 0,
|
||||
'teacher_count_10k': 0,
|
||||
'enrollment_count_10k': 0,
|
||||
'admission_count_10k': 0
|
||||
})
|
||||
|
||||
return result
|
||||
|
||||
|
||||
def print_education_data(data, year):
|
||||
"""打印教育数据"""
|
||||
print(f"\n===== {year}年云南省各学段教育数据统计（单位：学校总数-个，其余-万人）=====")
|
||||
print(f"{'学段':<8}{'学校总数':<10}{'教职工总数':<12}{'在校生总数':<12}{'招生总数':<12}")
|
||||
print("=" * 60)
|
||||
|
||||
for item in data:
|
||||
print(f"{item['education_stage']:<8}{item['school_count']:<10}{item['teacher_count_10k']:<12.2f}{item['enrollment_count_10k']:<12.2f}{item['admission_count_10k']:<12.2f}")
|
||||
|
||||
|
||||
def main():
|
||||
"""主函数"""
|
||||
# 默认查询2023年数据
|
||||
year = 2023
|
||||
|
||||
try:
|
||||
print(f"正在获取 {year} 年数据...")
|
||||
print(f"数据目录: {DATA_DIR}")
|
||||
|
||||
# 检查数据文件是否存在
|
||||
for file_path in [SCHOOL_COUNT_FILE, TEACHER_COUNT_FILE, ENROLLMENT_COUNT_FILE, ADMISSION_COUNT_FILE]:
|
||||
if os.path.exists(file_path):
|
||||
print(f"找到数据文件: {file_path}")
|
||||
else:
|
||||
print(f"警告：数据文件不存在: {file_path}")
|
||||
|
||||
# 获取数据
|
||||
education_data = get_education_data_by_year(year)
|
||||
|
||||
# 打印数据
|
||||
if education_data:
|
||||
print_education_data(education_data, year)
|
||||
else:
|
||||
print("未获取到任何数据")
|
||||
|
||||
except Exception as e:
|
||||
print(f"获取数据时发生错误: {e}")
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user