如何编写函数计算tree_data.csv中tree_area列的总面积？

下面提供两种实用的解决方案，分别适用于轻量数据和大数据量场景：

方法一：使用Python标准库csv（无需额外安装依赖） #

直接用Python自带的csv模块读取文件，按列名提取数据，处理空白单元格后累加：

import csv

def get_total_tree_area():
    total = 0.0
    # 打开CSV文件，指定编码避免乱码
    with open('tree_data.csv', 'r', newline='', encoding='utf-8') as f:
        # 用DictReader可以通过列名直接访问数据
        reader = csv.DictReader(f)
        for row in reader:
            # 去除字符串前后空格，处理空白单元格
            area_val = row['tree_area'].strip()
            # 空白则视为0，否则转为浮点数
            area = float(area_val) if area_val else 0.0
            total += area
    return total

# 调用示例
total_area = get_total_tree_area()
print(f"总面积：{total_area}")

方法二：使用pandas库（高效处理大数据） #

如果你的CSV文件数据量较大，pandas会更简洁高效，还能自动处理各种缺失值情况：

首先确保已经安装pandas：pip install pandas

然后编写代码：

import pandas as pd

# 只读取tree_area列，减少内存占用
df = pd.read_csv('tree_data.csv', usecols=['tree_area'])
# 将所有空白、缺失值填充为0，并转换为浮点数类型
df['tree_area'] = df['tree_area'].fillna(0).astype(float)
# 计算总和
total_area = df['tree_area'].sum()

print(f"总面积：{total_area}")

注意事项 #

• 确认CSV文件中的列名和代码里的tree_area完全一致，包括大小写和空格
• 如果空白单元格用"NA"、"null"等标识，可先替换再填充：df['tree_area'] = df['tree_area'].replace(['NA', 'null'], 0).fillna(0)
• 若CSV文件编码不是utf-8，需指定对应编码，比如encoding='gbk'

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