java – flatMap输入 – 尝试合并DoubleStreams

    final double[][] a = new double[][] { { 1, 2, 3 }, { 4, 5, 6 } };
    final int numRows = a.length;
    final int numCols = a[0].length;
    final double[] da = IntStream.range(0, numCols)
            .mapToObj(i -> IntStream.range(0, numRows).mapToDouble(j -> a[i][j])).flatMap(d -> d)
            .toArray(double[]::new);
    // da should be [1, 4, 2, 5, 3, 6]

错误是:无法推断flatMap的类型参数(函数>)

写这个的正确方法是什么?我已经尝试从地图中发出double [],但是也无法做到.谢谢一堆.

解决方法:

你的代码中有一个小错误:你在[i] [j]中混合了i和j.
改变这些指数你得到它:

double[] da = IntStream.range(0, numCols)
    .mapToObj(i -> IntStream.range(0, numRows).mapToDouble(j -> a[j][i]))
    .flatMapToDouble(d -> d)
    .toArray();

相反,flatMapToDouble(d – > d)也可以使用flatMapToDouble(Function.identity()).

上一篇:Java 8 中 Stream API 的这些奇技淫巧!你都 Get 到了吗?


下一篇:一眼看懂map和flatmap的区别