将这两个Java流合并为一个

是否有可能将方法中的两个流合并为一个,因此只有一条return语句行?该方法应计算给定的二维双精度数组中每一列的总和,该数组具有哪种结构无关紧要,但该数组至少具有一个元素,并且允许非空行。非常感谢。

    public static double[] sumUpCols(double[][] x) {
        double[][] a = Stream.of(x).sorted((a0, a1) -> a1.length - a0.length).toArray(double[][]::new);
        return IntStream.range(0, a[0].length)
                .mapToDouble(i -> IntStream.range(0, a[i].length).mapToDouble(j -> a[j][i]).sum()).toArray();
    }

    public static void main(String[] args) {
        double[][] x = { { 1 }, { 2, 3 }, { 4, 5, 6 } };
        System.out.println(Arrays.toString(sumUpCols(x)));
    }
评论