使用Java流创建数组中所有列表的列表[重复]

Question

我有一个 Node 数组，它们都有一个 Variable 列表：

Node[] arguments; // argument[i].vars() returns List<Variable>

我想创建一个包含所有变量的列表。我今天是这样做的：

List<Variable> allVars = new ArrayList<>();
for (Node arg : arguments) {
    allVars.addAll(arg.vars());
}

我可以使用流做同样的事情吗？

我已经尝试过了，但它返回给我一个List<List<Variable>>，而我想要一个带有所有列表元素的List<Variable>（使用addAll）：

List<List<Variable>> vars = Arrays.asList(arguments).stream()
                                  .map(Node::vars)
                                  .collect(Collectors.toList());

Answer 1

在调用collect之前，使用flatMap将Stream<List<Variable>>转换为Stream<Variable>：

List<Variable> vars = Arrays.asList(arguments).stream()
                            .map(Node::vars)
                            .flatMap(List::stream)
                            .collect(Collectors.toList());