Java 流删除前三个元素

Question

我有按学生id排序的对象列表：

List<Student> arraylist = new ArrayList<>();
arraylist.add(new Student(1, "Chaitanya", 26));
arraylist.add(new Student(2, "Chaitanya", 26));
arraylist.add(new Student(3, "Rahul", 24));
arraylist.add(new Student(4, "Ajeet", 32));
arraylist.add(new Student(5, "Chaitanya", 26));
arraylist.add(new Student(6, "Chaitanya", 26));

我想使用 stream 并仅删除学生 age 等于 26 的前三个元素。 你能帮我解决这个问题吗？

Answer 1

您可以将filter 和skip 用作：

List<Student> finalList = arraylist.stream()
        .filter(a -> a.getAge() == 26) // filters the students with age 26
        .skip(3) // skips the first 3 
        .collect(Collectors.toList());

这将导致列出年龄等于 26 岁的 Students，同时跳过前三个出现的此类学生。

---

另一方面，如果您只想从完整列表中排除这三个学生，您也可以这样做：

List<Student> allStudentsExcludingFirstThreeOfAge26 = Stream.concat(
            arraylist.stream().filter(a -> a.getAge() != 26),
            arraylist.stream().filter(a -> a.getAge() == 26).skip(3))
        .collect(Collectors.toList());

请注意，这可能会导致更改列表的原始顺序。

Answer 2

首先通过流 API 搜索您希望删除的元素：

List<Student> toRemove = arraylist.stream()
                                  .filter(x -> x.getAge() == 26)
                                  .limit(3)
                                  .collect(toList());

然后从源列表中删除：

toRemove.forEach(arraylist::remove);

如果您不想改变源代码，请先克隆它：

List<Student> resultSet = new ArrayList<>(arraylist);
List<Student> toRemove = arraylist.stream().filter(x -> x.getAge() == 26).limit(3)
                                  .collect(toList());
toRemove.forEach(resultSet::remove);

毕竟，使用迭代器可能会更好地实现：

int counter = 0;
for (Iterator<Student> it = arraylist.iterator(); it.hasNext();){        
      if(counter == 3) break;
      Student student = it.next();
      if (student.getAge() == 26){
          it.remove();
          counter++;
      }
}