Czy istnieje sposób drenażu części wektora na podstawie predykatu?

Question

Próbuję usunąć niektóre elementy z wektora, na podstawie predykatu i zbierając wynik. Oto (nie działa) przykład z oczekiwanego rezultatu:

let mut v: Vec<i32> = vec![1, 2, 3, 4, 5, 6]; 

let drained: Vec<i32> = v.iter().filter(|e| (*e) % 2 == 0).drain(..).collect(); 

assert_eq!(v, vec![1, 3, 5]); 
assert_eq!(drained, vec![2, 4, 6]); 

Skutkuje to błędem

error[E0599]: no method named `drain` found for type `std::iter::Filter<std::slice::Iter<'_, i32>, [closure@src/main.rs:4:45: 4:62]>` in the current scope 
--> src/main.rs:4:64 
    | 
4 |  let drained: Vec<i32> = v.iter().filter(|e| (*e) % 2 == 0).drain(..).collect(); 
    |                ^^^^^ 

Istnieje kilka alternatywnych Spojrzałem na żaden z nich nie wydaje się robić to, co chcę:

• Vec::retain usuwa elementy z wektora, ale nie oddać własność usuniętych elementów.

• v.drain(..).filter(condition).collect() zwraca poprawną wartość dla drained, ale opróżnia cały wektor.

Answer 1

Nie stabilny Rdza 1.20.0. Jest niestabilny nightly funkcja o nazwie drain_filter że robi dokładnie to, co chcesz:

#![feature(drain_filter)] 

fn main() { 
    let mut v: Vec<i32> = vec![1, 2, 3, 4, 5, 6]; 

    let drained: Vec<i32> = v.drain_filter(|&mut e| e % 2 == 0).collect(); 

    assert_eq!(v, vec![1, 3, 5]); 
    assert_eq!(drained, vec![2, 4, 6]); 
} 

jako stabilny obejścia, może być w stanie wykorzystać Iterator::partition, ale nie można ponownie użyć pamięci:

fn main() { 
    let v: Vec<i32> = vec![1, 2, 3, 4, 5, 6]; 

    let (drained, v): (Vec<_>, Vec<_>) = v.into_iter().partition(|&e| e % 2 == 0); 

    assert_eq!(v, vec![1, 3, 5]); 
    assert_eq!(drained, vec![2, 4, 6]); 
}