java 使用 lamba 表达式处理二维数组

编写数学优化算法时遇到一个问题：如何将一个二维数组满足一定条件的行，筛选出来并组成一个新的数组。

传统思路：

编写一个专门的方法，这个方法返回满足条件的新数组，但数组必须首选确定满足条件的行数，有点麻烦。

或者使用 ArrayList， 使用 add 来添加满足条件的行，但最后还得把 ArrayList 转化成数组，也有点麻烦。

新思路：

使用 lambda 表达式，将二维数组转化成 stream， 使用 filter 过滤条件， 然后将过滤后的数组转化为二维数组。

举例：从一个二维数组中筛选出第四个元素大于5的行。

import java.util.Arrays;


public class HelloWorld {
	
	
	public static void main(String[] args) {
			int[][] a = {{3, 6, 8, 10}, {4, 5, 6, 7}, {1, 2, 3, 4}}; // 筛选出第四个元素大于5的行		
						
			int[][] b = Arrays.stream(a).filter(p -> p[3] > 5).map(i -> Arrays.stream(i).toArray())
									.toArray(int[][] :: new);
			
			System.out.println(Arrays.deepToString(b));
	}
}

注： 必须首先用一个 map 将 每一个行引用转化成一维数组，然后在用第二个 toArray 结合 二维数组的构造函数，转化成二维数组。

输出结果：

[[3, 6, 8, 10], [4, 5, 6, 7]]