使用 Lambda 求平均值（双重存储为字符串）

Question

我有一个如下格式的对象

Map<String, List<ObjectDTO>> mapOfIndicators = new HashMap<>();

ObjectDTO 是

public class ObjectDTO {
 private static final long   serialVersionUID = 1L;

    private String iName;
    private String dName;
    private String countryName;
    private int year;

    //Since the value can be empty, using String instead of Double
    private String newIndex;
}

我正在尝试计算 newIndex 值的平均值，因为 getNewIndex 是字符串，所以基本上我无法做到这一点，并且可以有空字符串（值）。我无法将 null 值转换为 0，因为 newIndex 中将有 0 个值。

有没有更简单的方法来计算平均值？

例如。

mapOfIndicators = new HashMap<>();
List<IndicatorTableDTO> _tempValue = new ArrayList();
IndicatorTableDTO _iTDTO = new IndicatorTableDTO("iName", "dName", "countryName1", 2012, "1.00");
_tempValue.add(_iTDTO);
_iTDTO = new IndicatorTableDTO("iName", "dName", "countryName2", 2012, "");
_tempValue.add(_iTDTO);
_iTDTO = new IndicatorTableDTO("iName", "dName", "countryName3", 2012, "0.02");
_tempValue.add(_iTDTO);
_iTDTO = new IndicatorTableDTO("iName", "dName", "countryName4", 2012, "-0.25");
_tempValue.add(_iTDTO);
_iTDTO = new IndicatorTableDTO("iName", "dName", "countryName5", 2012, "0.10");
_tempValue.add(_iTDTO);

mapOfIndicators.put("Test1", _tempValue);

   _iTDTO = new IndicatorTableDTO("iName", "dName", "countryName1", 2012, "0.10");
   _tempValue.add(_iTDTO);
   _iTDTO = new IndicatorTableDTO("iName", "dName", "countryName2", 2012, "", "", "");
   _tempValue.add(_iTDTO);
   _iTDTO = new IndicatorTableDTO("iName", "dName", "countryName3", 2012, "", "", "");
   _tempValue.add(_iTDTO);
   _iTDTO = new IndicatorTableDTO("iName", "dName", "countryName4", 2012, "0.25", "", "");
   _tempValue.add(_iTDTO);
   _iTDTO = new IndicatorTableDTO("iName", "dName", "countryName5", 2012, "1.10", "", "");
   _tempValue.add(_iTDTO);

   mapOfIndicators.put("Test2", _tempValue);

更新 01：我们可以跳过空字符串/空值；平均值是每个 countryName，所以我想找到相同 contryName 的 newIndex 的平均值，并最终能够得到一个 Map 作为最终结果。

Answer 1

未经测试，但应该可以解决问题（或至少引导您找到解决方案）：

Map<String, Double> averageByCountryName = 
    map.values()
       .stream()
       .flatMap(list -> list.stream())
       .filter(objectDTO -> !objectDTO.getNewIndex().isEmpty())
       .collect(Collectors.groupingBy(
            ObjectDTO::getCountry,
            Collectors.averagingDouble(
                objectDTO -> Double.parseDouble(objectDTO.getNewIndex())));

Answer 2

如果您想计算这些值的平均值并将它们收集在与源映射具有相同键的Map 中，您可以这样做：

Map<String, Double> averages=mapOfIndicators.entrySet().stream()
    .collect(Collectors.toMap(Map.Entry::getKey, e->e.getValue().stream()
       .map(ObjectDTO::getNewIndex).filter(str->!str.isEmpty())
       .mapToDouble(Double::parseDouble).average().orElse(Double.NaN)));

由于ObjectDTO.newIndex 被声明为private，我假设有一个方法getNewIndex 来获取它的值。我还提供了默认值 Double.NaN 作为问题中未指定的值列表中没有非空 Strings 的情况的行为。

---

如果您想对不同的键进行平均，因为您现在更精确地定义了您的问题，代码可能如下所示：

Map<String, Double> averages=mapOfIndicators.values().stream()
    .flatMap(Collection::stream)
    .filter(objectDTO -> !objectDTO.getNewIndex().isEmpty())
    .collect(Collectors.groupingBy(ObjectDTO::getCountryName,
        Collectors.mapping(ObjectDTO::getNewIndex,
            Collectors.averagingDouble(Double::parseDouble))));

嗯，这类似于JB Nizet’s answer，它在第一次猜测时就让你的意图正确......