Rust语言从入门到精通系列之Iterator迭代器深入详解
在Rust语言中，迭代器（Iterator）是一种极为重要的数据类型，它们用于遍历集合中的元素。Rust中的大多数集合类型都可转换为一个迭代器，使它们可以进行遍历，这包括数组、向量、哈希表等。
使用迭代器可以让代码更加简洁优雅，并且可以支持一些强大的操作，例如过滤、映射和折叠等。
在本文中，我们将探讨Rust语言中的迭代器的相关知识，并且以我们的老朋友Animal为例，提供相关的示例代码。
熟悉Java的Stream和Lambda的同学，学习本章节时，会格外的感觉“亲切”。
迭代器的基本概念
迭代器是什么？
在Rust中，迭代器是一个实现了Iterator trait的类型。该trait定义了一组行为，用于支持遍历集合中的元素。通过实现Iterator trait，类型可以被转换为一个迭代器，从而支持Iterate的操作。
Iterator trait
Iterator trait 定义了迭代器的核心行为，它包含了next方法和一些其他方法。next方法返回集合中下一个元素的Option值，直到集合中所有的元素都被遍历完毕，返回None。
除了next方法之外，Iterator trait 还定义了其他许多有用的方法，比如map、filter等，这些方法可以对迭代器中的元素进行操作和转换。
pub trait Iterator {
type Item;
fn next(&mut self) -> Option<Self::Item>;
// 多种内置实现方法, skip, map, reduce, collect
// 和Java中的Stream内置方法非常类似.
}
Animal示例
接下来我们探讨实现一个Animal迭代器，Animal实现Iterator trait，使其可以通过迭代器遍历Animal的各个属性。 以下是Animal类型的定义：
#[derive(Debug)]
struct Animal {
name: String,
age: u32,
kind: String,
i:i32,
}
我们可以在Animal上实现Iterator trait，使其可以通过for循环进行迭代。
impl Iterator for Animal {
type Item = String;
fn next(&mut self) -> Option<Self::Item> {
let next_attribute = match self.i {
0 => Some(self.name.clone()),
1 => Some(self.age.to_string()),
2 => Some(self.kind.clone()),
_ => None,
};
self.i += 1;
next_attribute
}
}
此时，我们已经将我们的类型转换为迭代器，我们就可以在它上面调用各种Iterator trait 的方法。例如，我们可以使用for循环遍历Animal对象的每一个属性：
#[derive(Debug)]
struct Animal {
name: String,
age: u32,
kind: String,
i:i32,
}
impl Iterator for Animal {
type Item = String;
fn next(&mut self) -> Option<Self::Item> {
let next_attribute = match self.i {
0 => Some(self.name.clone()),
1 => Some(self.age.to_string()),
2 => Some(self.kind.clone()),
_ => None,
};
self.i += 1;
next_attribute
}
}
fn main() {
let mut animal = Animal {
name: "Tom".to_string(),
age : 15,
kind: "cat".to_string(),
i : 0
};
println!("Name: {}", animal.next().unwrap());
println!("Age: {}", animal.next().unwrap());
println!("Kind: {}", animal.next().unwrap());
}
// 输出结果：
// Name: Tom
// Age: 15
// Kind: cat
在上述代码中，我们定义了一个Animal类型的Iterator，并定义了一个名为i的内部状态变量。该变量用于追踪遍历的进度，并决定下一个迭代器值的内容。最终成功打印了animal的全部信息。
下面继续我们的学习，定一个Animal向量并遍历打印每一个Animal的所有属性：
fn print_all_attributes(animals: Vec<Animal>) {
for mut animal in animals {
println!("Name: {}", animal.next().unwrap());
println!("Age: {}", animal.next().unwrap());
println!("Kind: {}", animal.next().unwrap());
}
}
fn main() {
let animals = vec![Animal {
name: "Tom".to_string(),
age : 15,
kind: "cat".to_string(),
i : 0
}];
print_all_attributes(animals);
}
// 输出结果：
// Name: Tom
// Age: 15
// Kind: cat
在上述代码中，我们使用for循环来遍历所有的Animal对象，并逐一打印它们的属性。
迭代器的常见用法
map方法
map方法是Iterator trait 中非常重要的一个方法，它可以让我们对迭代器中的每一个元素进行转换操作，并返回新的迭代器。例如：
fn main() {
let animals = vec![Animal {
name: "Tom".to_string(),
age : 15,
kind: "cat".to_string(),
i : 0
}, Animal {
name: "Jerry".to_string(),
age : 7,
kind: "mouse".to_string(),
i : 0
}];
let list: Vec<String> = animals
.into_iter()
.map(|ani| ani.name.clone())
.collect();
println!("{:?}", list)
}
// 输出 ["Tom", "Jerry"]
上述代码中，我们定义了一个包含2个的向量animals，并使用iter方法将其转换为一个迭代器。然后，我们使用map方法对这个迭代器中的Animal的name操作，返回一个新的迭代器，并使用collect方法将其转换为向量list。
filter方法
假设我们现在想寻找年龄大于等于3岁的动物，我们可以使用filter方法来实现。
fn main() {
let animals = vec![Animal {
name: "Tom".to_string(),
age : 15,
kind: "cat".to_string(),
i : 0
}];
let filtered_animals: Vec<Animal> = animals
.into_iter()
.filter(|animal| animal.age >= 3)
.collect();
println!("{:?}", filtered_animals)
}
// 输出结果：
// [Animal { name: "Tom", age: 15, kind: "cat", i: 0 }]
在上述代码中，我们使用into_iter方法将Animal向量转换为迭代器，并使用filter方法过滤其中年龄大于等于3岁的动物，最终返回一个新的Animal向量。
enumerate方法
enumerate方法会将一个迭代器中的元素和它们的索引配对，并返回一个新的迭代器。例如：
fn main() {
let animals = vec![Animal {
name: "Tom".to_string(),
age : 15,
kind: "cat".to_string(),
i : 0
}, Animal {
name: "Jerry".to_string(),
age : 7,
kind: "mouse".to_string(),
i : 0
}];
for (i, animal) in animals.iter().enumerate() {
println!("{}: {:?}", i, animal);
}
}
// 输出：
// 0: Animal { name: "Tom", age: 15, kind: "cat", i: 0 }
// 1: Animal { name: "Jerry", age: 7, kind: "mouse", i: 0 }
上述代码中，我们定义了一个包含2个Animal的向量animals，并使用iter方法将其转换为一个迭代器。然后，我们使用enumerate方法将每Animal与其索引配对，并在for循环中打印出来。
flat_map方法
flat_map方法是Iterator trait 中比较少见的方法之一，它可以用于将嵌套的迭代器展开为单个迭代器。例如：
#[derive(Debug, Clone)]
struct Animal {
name: String,
age: u32,
kind: String,
i: i32,
}
fn main() {
let cat = Animal {
name: "Tom".to_string(),
age: 15,
kind: "cat".to_string(),
i: 0,
};
let mouse = Animal {
name: "Jerry".to_string(),
age: 7,
kind: "mouse".to_string(),
i: 0,
};
let animals = vec![vec![cat], vec![mouse]];
let list: Vec<Animal> = animals.iter().flat_map(|x| x.iter().cloned()).collect();
println!("{:?}", list)
}
// 输出 [Animal { name: "Tom", age: 15, kind: "cat", i: 0 }, Animal { name: "Jerry", age: 7, kind: "mouse", i: 0 }]
上述代码中，我们定义了一个二维向量animals，并使用iter方法将它转换为迭代器。然后，我们使用flat_map方法将它展开为一个一维的迭代器，并使用collect方法将其转换为向量list。
zip方法
如果我们需要同时遍历两个向量，我们可以使用zip方法进行配对。
fn main() {
let names = vec!["Tom", "Jerry", "Bob"];
let ages = vec![3, 4, 5];
for (name, age) in names.iter().zip(ages.iter()) {
println!("{} is {} years old.", name, age);
}
}
// 输出结果：
// Tom is 3 years old.
// Jerry is 4 years old.
// Bob is 5 years old.
上述代码中，我们使用iter方法将names和ages向量转换为迭代器，并使用zip方法对它们进行配对。对于每一对元素，我们调用println!函数并打印它们。
fold方法
fold方法在Rust中也十分重要，它可以接受一个初始值和一个闭包，遍历迭代器中的每一个元素，并将它们合并成单个值。例如：
fn main() {
let cat = Animal {
name: "Tom".to_string(),
age: 15,
kind: "cat".to_string(),
i: 0,
};
let mouse = Animal {
name: "Jerry".to_string(),
age: 7,
kind: "mouse".to_string(),
i: 0,
};
let animals = vec![cat, mouse];
let sum = animals.iter().fold(0, |t, ani| t + ani.age );
println!("{}", sum)
}
// 输出 22
上述代码中，我们定义了一个包含2个Animal的向量animals，并使用iter方法将其转换为一个迭代器。然后，我们使用fold方法对这个迭代器中的age进行累加，并返回结果sum。
结论
迭代器是Rust语言中非常重要的数据类型，它们用于遍历集合中的元素，并支持各种操作。在本教程中，我们探讨了迭代器的基本概念和常见用法，以Animal为例子，提供了相应的演示代码。希望读者能够掌握Rust迭代器的相关内容，并且在实际编程中得到应用。
以上就是Rust语言从入门到精通系列之Iterator迭代器深入详解的详细内容，更多关于Rust Iterator迭代器的资料请关注脚本之家其它相关文章！
相关文章
这篇文章主要介绍了Rust语言中的String和HashMap使用示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-10-10
在 Rust 中,标准库提供了一组核心功能,以帮助开发者执行常见的编程任务,这个路径树可以作为参考,帮助你更好地理解 Rust 标准库的结构和模块之间的关系,本文介绍 Rust 标准库的结构,并提供相应的 use 路径,感兴趣的朋友一起看看吧2024-05-05
Rust开发WebAssembly在Html和Vue中的应用小结(推荐)
这篇文章主要介绍了Rust开发WebAssembly在Html和Vue中的应用,本文将带领大家在普通html上和vue手脚架上都来运行wasm的流程，需要的朋友可以参考下2022-08-08
在Rust中,字符串方法主要位于标准库的std::string模块中,这些方法可以帮助我们处理字符串的常见操作,本文主要介绍了Rust中字符串String集合的具有使用,具有一定的参考价值,感兴趣的可以了解一下2024-04-04
web server 中主要的两个协议是 http 和 tcp,tcp 是底层协议,http 是构建在 tcp 之上的,本篇文章重点给大家介绍rust 创建多线程web server的详细过程,感兴趣的朋友跟随小编一起看看吧2023-11-11
在Rust语言中, 每一个引用都有其生命周期, 通俗讲就是每个引用在程序执行的过程中都有其自身的作用域, 一旦离开其作用域, 其生命周期也宣告结束, 值不再有效,这篇文章主要介绍了Rust编程中的生命周期,需要的朋友可以参考下2023-11-11
Rust中的宏是一种强大的工具，可以帮助开发人员编写可重用、高效和灵活的代码，这篇文章主要介绍了Rust中的宏：声明宏和过程宏,需要的朋友可以参考下2023-04-04
这篇文章主要介绍了Rust中的Copy和Clone及区别对比分析，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2023-04-04
字符串有两种表现形式,一种是基本类型,表示字符串的切片,以&str表示,另一种是可变的string类型,下面这篇文章主要给大家介绍了关于Rust字符串字面值的相关资料,需要的朋友可以参考下2022-04-04
异步编程在 Rust 中的地位非常高，很多 crate 尤其是多IO操作的都使用了 async/await，这篇文章主要介绍了Rust 如何实现 async/await,需要的朋友可以参考下2022-09-09
最新评论