Python基于条件计算CSV列平均值时遇除零错误求助

你遇到的除零错误核心原因是：如果分两次读取文件，第一次读完后文件指针会停在文件末尾，第二次读取时没有数据，导致其中一个性别的统计数量为0，触发除法错误。根本不需要重复打开文件，只需要一次遍历文件，同时统计男女的体重总和与人数就能解决。

关键解决思路 #

一次遍历CSV的每一行，同时维护四个变量：男性体重总和、男性人数、女性体重总和、女性人数，遍历完成后再分别计算平均值。这样既避免了文件指针问题，又提升了效率。

示例代码 #

import csv

# 初始化统计变量
male_total = 0.0
male_count = 0
female_total = 0.0
female_count = 0

# 用with语句自动管理文件生命周期，无需手动关闭
with open('your_data.csv', 'r', newline='', encoding='utf-8') as csvfile:
    # 假设CSV有表头，字段名比如是'gender'和'initial_weight'，根据实际调整
    reader = csv.DictReader(csvfile)
    for row in reader:
        # 统一性别格式，避免大小写、空格导致的统计遗漏
        gender = row['gender'].strip().lower()
        # 处理体重数据，跳过无效值
        try:
            weight = float(row['initial_weight'])
        except ValueError:
            continue
        
        # 累加对应性别的数据
        if gender == 'male':
            male_total += weight
            male_count += 1
        elif gender == 'female':
            female_total += weight
            female_count += 1

# 计算平均时先判断计数是否为0，避免除零错误
male_avg = male_total / male_count if male_count > 0 else 0.0
female_avg = female_total / female_count if female_count > 0 else 0.0

print(f"男性初始平均体重: {male_avg:.2f}")
print(f"女性初始平均体重: {female_avg:.2f}")

需要注意的细节 #

• 确认CSV的表头字段名和代码中一致，比如如果表头是'Gender'而非'gender'，要对应修改row['Gender']
• 如果数据中性别用缩写（比如'M'/'F'）或中文（'男'/'女'），要调整gender == 'male'这类判断条件
• 加入try-except处理无效体重数据，防止因单元格为空或非数字导致程序中断

内容的提问来源于stack exchange，提问作者P0ffee