将嵌套的 For 循环转换为流

Question

我在理解流时遇到了一些问题。我环顾四周，似乎找不到与我的用例匹配的示例。

我有一个现有的嵌套 for 循环：

List<ObjectB> objectBs = new ArrayList<ObjectB>();
for (ObjectA objA: objectAList) {
    for (ObjectB objB: objA.getObjectBList()) {
        if (objB.getNumber() != 2) {
            objectBs.add(objB);
        }
    }
}

大量示例展示了如何将 objB.getNumber() 添加到列表中，而不是 objB。

Answer 1

您可以使用flatMap 获取所有ObjectB 实例的Stream<ObjectB> 并过滤所需数量的ObjectB：

List<ObjectB> objectBs = 
    objectAList.stream()
               .flatMap (a -> a.getObjectBList().stream())
               .filter (b -> b.getNumber() != 2)
               .collect (Collectors.toList());