在 Java 中流式传输 2D 数组的一部分

Question

假设你有一个矩阵

[[1,2,5],[3,4,6]]

并且您需要一起处理每行的前两个元素

[1,2,3,4]

。您不想将其存储在内存中或使用嵌套循环。有没有办法使用 Java 中的流来实现这一点？

Answer 1

这是例子

int[][] arr = {{1, 2, 5}, {3, 4, 6}};

Arrays.stream(arr)
        .flatMap(a -> Arrays.stream(a).limit(2).mapToObj(x -> x))
        .forEach(x -> System.out.println(x));