使用流api根据条件将列表拆分为子列表

Question

我有一个具体的问题。也有一些类似的问题，但这些问题要么与Python有关，而不是与Java有关，要么即使问题听起来类似，需求也不同。

我有一份价值观清单。

List1 = {10, -2, 23, 5, -11, 287, 5, -99}

最后，我想根据列表的值对列表进行拆分。我的意思是，如果值大于零，它将保留在原始列表中，负值列表中的相应索引将设置为零。如果该值小于零，它将转到负值列表，原始列表中的负值将替换为零。

结果列表应该是这样的；

List1 = {10, 0, 23, 5, 0, 287, 5, 0}
List2 = {0, -2, 0, 0, -11, 0, 0, -99}

有没有办法解决这个流api在Java？

Answer 1

正如shmosel在评论中指出的，您需要使用streams进行两次迭代：

List<Integer> list = Arrays.asList(10, -2, 23, 5, -11, 287, 5, -99);
List<Integer> positives = list.stream().map(i -> i < 0 ? 0 : i).collect(Collectors.toList());
List<Integer> negatives = list.stream().map(i -> i < 0 ? i : 0).collect(Collectors.toList());

如果您的列表是可修改的，则可以在一个流中进行所有操作。这不比for循环好多少

List<Integer> list = Arrays.asList(10, -2, 23, 5, -11, 287, 5, -99);
List<Integer> list2 = new ArrayList<>();

IntStream.range(0, list.size()).forEach(i -> {
   int j;
   if ((j = list.get(i)) < 0) {
       list2.add(j);
       list.set(i, 0);
   } else {
       list2.add(0);
   }}); 

Answer 2

如果要在单一Stream操作中执行此操作，则需要自定义收集器：

List<Integer> list = Arrays.asList(10, -2, 23, 5, -11, 287, 5, -99);

List<List<Integer>> result = list.stream().collect(
    () -> Arrays.asList(new ArrayList<>(), new ArrayList<>()),
    (l,i) -> { l.get(0).add(Math.max(0, i)); l.get(1).add(Math.min(0, i)); },
    (a,b) -> { a.get(0).addAll(b.get(0)); a.get(1).addAll(b.get(1)); });

System.out.println(result.get(0));
System.out.println(result.get(1));

Answer 3

Map<Boolean, List<Integer>> results=
  List1.stream().collect(Collectors.partitioningBy( n -> n < 0));

我认为这个更漂亮，更容易阅读。（然后，您可以从地图中获得否定和非否定列表。）