主页 / 问题 / 941843
Accepted
Anton Sorokin
Asked:2020-02-07 15:23:07 +0000 UTC2020-02-07 15:23:07 +0000 UTC

如何使用 Stream API 从列表中获取每个第 n 个元素？

  • 772

假设我有一个这样的列表：

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

是否可以使用 Java Stream API 包装此列表中的每个第二个元素以获得这样的列表？

[1, 3, 5, 7, 9]

或者也许每三个元素？

[1, 4, 7, 10]

基本上，我正在寻找这样的函数来从流中获取每个第 n 个元素：

List<Integer> list = Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);
List<Integer> list2 = list.stream().takenth(3).collect(Collectors.toList());
System.out.println(list2);
// => [1, 4, 7, 10]
java

2 个回答

  1. Best Answer

    引入 Java 线程的主要动机之一是允许并行操作。这导致要求 Java Streams 操作（例如mapand filter）独立于流中元素的位置或其周围的元素。这样做的好处是可以轻松拆分线程以进行并行处理。缺点是某些操作的复杂性。

    因此，没有简单的方法来执行诸如获取每第 n 个项目或将每个项目与所有先前项目的总和匹配的事情。

    满足您的要求的最简单方法是使用列表索引：

    List<String> list = ...;
return IntStream.range(0, list.size())
    .filter(n -> n % 3 == 0) //3 - каждый 3-ий элемент
    .mapToObj(list::get)
    .collect(Collectors.toList());

    谢尔盖·戈尔诺斯塔耶夫的注意事项：

    解决方案是有效的，但我必须说，这种流的使用是一种反模式。不希望使用流外部的数据。访问变量的想法一出现，您就应该保持警惕并思考“我是在错误地使用流，还是流根本不适合这里？”

    使用Stream.iterate 

    List<Integer> list = Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);

int skip = 3;
int size = list.size();
// Limit to carefully avoid IndexOutOfBoundsException
int limit = size / skip + Math.min(size % skip, 1);

List<Integer> result = Stream.iterate(0, i -> i + skip)
    .limit(limit)
    .map(list::get)
    .collect(Collectors.toList());

System.out.println(result); // [1, 4, 7, 10]

    使用Guava库（GitHub）：

    Streams
    .mapWithIndex(stream, SimpleImmutableEntry::new)
    .filter(entry -> entry.getValue() % 3 == 0)
    .map(Entry::getKey)
    .collect(Collectors.toList());

    使用AbacusUtil库：

    Stream.of(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
        .filter(MutableInt.of(0), (e, idx) -> idx.getAndDecrement() % 3 == 0)
        .println();

    使用jOOλ库及其方法zipWithIndex()：

    System.out.println(
Seq.of(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
   .zipWithIndex()             // This produces a Tuple2(yourvalue, index)
   .filter(t -> t.v2 % 3 == 0) // Filter by the index
   .map(t -> t.v1)             // Remove the index again
   .toList()
);
    • 5 
  2. List<String> original = List.of("a", "b", "c", "d", "e", "f", "g", "h", "i", "j");
List<Integer> intList;
{
    AtomicInteger index = new AtomicInteger(0);
    intList = original.stream()
        .filter(e -> ((index.getAndIncrement()) % 2 == 0))
        .map(s -> s.charAt(0) - 'a' + 1) // для примера поинтереснее
        .collect(toList());
}
System.out.println(intList); // [1, 3, 5, 7, 9]
    • 0

相关问题