MongoDB Java中分离键值对的值并实现求和、求平均值的方法

嘿，我看你需要从MongoDB集合里只提取pr字段的值，然后计算总和和平均值对吧？我帮你把代码补全并优化了，直接能用：

核心思路很简单：先通过投影只获取pr字段，遍历所有符合条件的文档提取pr值，累加求和后再计算平均值。

private static void calculatePrSumAndAverage(DBCollection collection) {
    // 构建查询条件：保留你原有的st=17过滤规则
    BasicDBObject query = new BasicDBObject("st", 17);
    
    // 投影配置：只返回pr字段，排除默认的_id字段
    BasicDBObject projection = new BasicDBObject("pr", 1);
    projection.put("_id", 0);
    
    // 执行查询，获取结果游标
    DBCursor cursor = collection.find(query, projection);
    
    int totalPseudorange = 0;
    int recordCount = 0;
    
    try {
        // 遍历所有查询到的文档
        while (cursor.hasNext()) {
            BasicDBObject currentDoc = (BasicDBObject) cursor.next();
            // 提取pr字段的值（这里假设pr是整数类型，若为其他类型请调整转换方式）
            Integer prValue = currentDoc.getInt("pr");
            
            // 空值判断，避免字段缺失或为null时抛出异常
            if (prValue != null) {
                totalPseudorange += prValue;
                recordCount++;
            }
        }
        
        // 计算并输出结果，注意避免除以0的情况
        if (recordCount > 0) {
            double averagePr = (double) totalPseudorange / recordCount;
            System.out.println("伪距总和: " + totalPseudorange);
            System.out.println("伪距平均值: " + averagePr);
        } else {
            System.out.println("没有找到符合条件的记录哦");
        }
    } finally {
        // 务必关闭游标，释放数据库连接资源
        cursor.close();
    }
}

几个关键注意点： #

• 我把你原来的多个whereQuery合并成了一个query对象，逻辑更清晰，避免冗余
• 增加了空值判断，如果某些文档里没有pr字段或者值为null，不会影响整体计算
• 最后加了记录数量判断，防止没有查到数据时出现除以0的错误
• 游标一定要在finally块里关闭，不然会造成数据库资源泄漏

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