将单个对象和对象列表合并为Java中的单个对象列表

时间:2018-12-14 06:18:19

标签: java list arraylist java-8 java-stream

给出对象列表

// ["A", ["B", null, "C"], "D", ["E"], [], null, "F"];
List<Object> objects = Arrays.asList("A", Arrays.asList("B", null, "C"), "D", Arrays.asList("E"), Arrays.asList(), null, "F");

是否有一种简单的方法可以将非null值组合到单个列表["A", "B", "C", "D", "E", "F"]中?

有些人坚持认为帖子必须显示尝试过的内容

objects.stream().map(o -> {
        if (o instanceof List) {
            return ((List) o).stream();
        } else return objects.stream();
    }).filter(Objects::nonNull).collect(Collectors.toList()));

显然上述方法无效。

1 个答案:

答案 0 :(得分:7)

如果只有两层嵌套,则列表中的较深列表不会递归地进行, 那么这是一种扁平化方法:

List<String> flat = objects.stream()
  .filter(Objects::nonNull)
  .flatMap(v -> {
    if (v instanceof String) {
      return Stream.of((String) v);
    }
    return ((List<String>) v).stream().filter(Objects::nonNull);
  })
  .collect(Collectors.toList());

如果可以递归进行更深层次的嵌套, 那么您可以将上述内容包装在一个函数中,然后递归地应用它(感谢@Holger使其更加紧凑):

List<String> flatten(List<?> objects) {
  return objects.stream()
    .filter(Objects::nonNull)
    .flatMap(v -> v instanceof String
      ? Stream.of((String) v)
      : flatten((List<?>) v).stream())
    .collect(Collectors.toList());
}