使用 Java 8 流 API 减少映射

hashmap java java-8 java-stream

2022-09-04 07:57:21

我有以下形式的地图：

Map<Integer, Map<String,Double>> START

设 INNER 是内在映射，即

Map<String,Double>

例如，我想在新地图中减少START地图

Map<Integer, Double> END

它们具有相同的键，但值不同。特别是，对于每个键，我希望新的 Double 值是相应键的 INNER 映射中值的总和。

如何使用JAVA 8的STREAM API来实现这一点？

谢谢大家。

编辑：示例地图是

------------------------------
|      |  2016-10-02   3.45   |
| ID1  |  2016-10-03   1.23   |
|      |  2016-10-04   0.98   |
------------------------------
|      |  2016-10-02   1.00   |
| ID2  |  2016-10-03   2.00   |
|      |  2016-10-04   3.00   |
------------------------------

e 我想要一张如下的新地图：

--------------------------------
|      |                       |
| ID1  |  SUM(3.45,1.23,0.98)  |
|      |                       |
--------------------------------
|      |                       |
| ID2  |  SUM(1.00,2.00,3.00)  |
|      |                       |
--------------------------------

答案 1

它将为您工作

    Map<Integer, Double> collect = START.entrySet()
        .stream()
        .collect(
            Collectors.toMap(
                Map.Entry::getKey, 
                e -> e.getValue()
                      .values()
                      .stream()
                      .reduce(0d, (a, b) -> a + b)
                )
        );

答案 2

这应该是一个很好的例子：

public class Main {

    public static void main(final String[] args) {
        final Map<Integer, Map<String, Double>> tmp = new HashMap<>();
        tmp.put(1, new HashMap<String, Double>() {{
            put("1", 3.45);
            put("2", 1.23);
            put("3", 0.98);
        }});
        tmp.put(2, new HashMap<String, Double>() {{
            put("1", 1.00);
            put("2", 2.00);
            put("3", 3.00);
        }});

        System.out.println(tmp.entrySet().stream()
                .collect(
                    Collectors.toMap(Map.Entry::getKey, 
                    data -> 
                        data.getValue()
                            .values().stream()
                            .mapToDouble(Number::doubleValue).sum())));
    }

}

输出将是，所有这一切都是获取映射的入口集，获取它的流并收集到内部映射值的新映射总和。{1=5.66, 2=6.0}