java - 使用包含列表的流过滤数据？

Question

elementList是所有元素的列表

class Element{  
   private ArrayList<Shape> shapeList = new ArrayList<Shape>();
}

...

class Shape{
   private String color;
   private String shape;
}

尝试过滤元素列表，其中每个元素都包含不同形状的列表。

    List<Shape> roundShapes = elementList.stream()  
            .filter(x -> x.getShapeList()(})    

 //not sure what to have here.
 //I need to loop/stream the data that i get form getShapeList(). How to do this?   
 //A stream inside a stream?

            .collect(Collectors.toList());

例如，找到一个红色且圆形的形状。问题是我不能直接过滤它，因为数据位于另一个列表中。

我可以循环所有元素并将每个 shapeList 添加到一个大列表中，然后流式传输该列表。但这需要一个 for 循环并迭代每个循环，我想使用流而不是 if for 循环。

Answer 1

您需要展平 map ，然后迭代形状。
将 shape.getShape().equals("Round") 替换为所需形状的过滤条件。

List<Shapes> roundShapes = elementList.stream().flatMap(element -> element.getShapeList().stream())
                    .filter(shape -> shape.getShape().equals("Round")).collect(Collectors.toList());