如何使用生成器函数初始化Rust向量?

时间:2017-12-29 10:50:32

标签: rust

在语言之间跳跃可能会很痛苦。一种语言的习语"感觉良好"一个人开始在其他语言中寻找相同的习语。

在F#中,有一种借助生成器函数初始化数组的方法。 Rust standard library documentation about vectors。现在,我跳到Rust一段时间,我想知道是否有类似的设施或者我是否必须创建自己这样的设施。

学习enter image description here,我找不到任何类似于我正在寻找的东西。

// Looking for something similar to:
Vec<T>::init(n : usize, generator : F) -> Vec<T> 
    where F: Fn(usize) -> T {
    // ...
}

也许它通过迭代器在Rust中的工作方式不同。但我必须承认,Rust迭代器(以及它们无数的味道)对我的简单思想来说仍然有点模糊。

3 个答案:

答案 0 :(得分:14)

您可以使用range map,然后收集结果。

F# docs的示例相似:

let my_vector : Vec<i32> = (1..11).map(|x| x*x).collect();

选中此live example

答案 1 :(得分:4)

虽然@Netwave提供了答案,但我指出了一个使用它的解决方案,并提供了更多的可重用性和可读性。

  1. 定义通用函数playground

    a
    Hello
    
  2. 定义一个特征,根据需要实现它并使用它,playground

    fn initialize<T>(count: usize, f: fn(usize) -> T) -> Vec<T> {
        (0..count).map(f).collect()
    }
    
    fn main() {
        let my_vector = initialize(10, |i| i as i32);
        for e in my_vector {
             println!("{}", e);
        }
    }
    

    more general way

    trait Initializer<T> {
        fn initialize(count: usize, f: fn(usize) -> T) -> Vec<T> {
            (0..count).map(f).collect()
        }
    }
    
    impl<T> Initializer<T> for Vec<T> {}
    
    fn main() {
        let my_vector = Vec::initialize(10, |i| i as i32);
        for e in my_vector {
            println!("{}", e);
        }
    }
    
  3. 写一个宏,playground

    trait Initializer<T, U> {
        fn initialize(count: usize, f: fn(usize) -> U) -> T;
    }
    
    impl<T: std::iter::FromIterator<U>, U> Initializer<T, U> for T {
        fn initialize(count: usize, f: fn(usize) -> U) -> T {
            (0..count).map(f).collect::<T>()
        }
    }
    
    fn main() {
        let my_vector = Vec::initialize(10, |i| i as i32);
        for e in my_vector {
            println!("{}", e);
        }
    }
    
  4. 基础仍然是@ Netwave的回答。你问过你想要的东西:

    macro_rules! vec_init {
        ($count: expr, $f: expr) => { (0..$count).map($f).collect() }
    }
    
    fn main() {
        let my_vector: Vec<i32> = vec_init!(10, |i| i as i32);
    }
    

    第二项中确实存在此代码。

答案 2 :(得分:0)

这次聚会来晚了,但是:resize_with