python 判断地表水质等级算法

目的

用 python实现《地表水环境质量标准》（GB 3838-2002）水质等级判断。

标准要求

水质标准限制如下图，其中pH上限是9，下限是6，超过上下限都是超标。溶解氧要求的是下限，其他均为上限。

Python 代码片段

limits = {
    "高锰酸盐指数": [2, 4, 6, 10, 15],
    "氨氮": [0.15, 0.5, 1, 1.5, 2.0],
    "总磷": [0.02, 0.1, 0.2, 0.3, 0.4],
    "化学需氧量": [15, 15, 20, 30, 40],
    "五日生化需氧量": [3, 3, 4, 6, 10],
    "氟化物": [1, 1, 1, 1.5, 1.5],
    "pH": [6, 9],  # pH的特殊处理，下限为6，上限为9
    "溶解氧": [7.5, 6, 5, 3, 2]  # 溶解氧的特殊处理，只有下限
    
}
def water_quality_level(monitoring_factor, value, limits):
    if monitoring_factor == "pH":
        if value < limits[monitoring_factor][0] or value > limits[monitoring_factor][1]:
            return "超标"
        else:
            return "正常"
    elif monitoring_factor == "溶解氧":
        if value >= limits[monitoring_factor][0]:
            return "Ⅰ"
        elif value >= limits[monitoring_factor][1]:
            return "Ⅱ"
        elif value >= limits[monitoring_factor][2]:
            return "Ⅲ"
        elif value >= limits[monitoring_factor][3]:
            return "Ⅳ"
        elif value >= limits[monitoring_factor][4]:
            return "Ⅴ"
        else:
            return "劣Ⅴ"
    else:
        if value <= limits[monitoring_factor][0]:
            return "Ⅰ"
        elif value <= limits[monitoring_factor][1]:
            return "Ⅱ"
        elif value <= limits[monitoring_factor][2]:
            return "Ⅲ"
        elif value <= limits[monitoring_factor][3]:
            return "Ⅳ"
        elif value <= limits[monitoring_factor][4]:
            return "Ⅴ"
        else:
            return "劣Ⅴ"

# 示例调用
print(water_quality_level("pH", 5.5, limits))  # 应该输出 "超标"
print(water_quality_level("溶解氧", 6.5, limits))  # 应该输出 "Ⅱ"

BY

纯个人经验，如有帮助，请收藏点赞，如需转载，请注明出处。
微信公众号：环境猫 er
CSDN : 细节处有神明
个人博客： https://maoyu92.github.io/