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

2023年3月10日 148次阅读 来源: 心态与做事习惯决定人生高度

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

传统思路：

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

或者使用 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]]

    原文作者：心态与做事习惯决定人生高度
    原文地址: https://blog.csdn.net/robert_chen1988/article/details/78726060
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞