Rust 集合类型-连式操作及类型转换

Rust是一种系统编程语言，以其性能、可靠性和强大的内存安全保证而闻名。Rust的一个关键特性是它的集合类型，它提供了有效和安全的方法来管理和操作数据。在本文中，我们将探讨一些最常用的Rust集合及其用例，以及常用操作和类型转换。

链式操作

链式操作指的是对集合应用一系列操作或转换。在Rust中，Iterator特性提供了执行链操作的各种方法。这些方法适用于大多数集合类型，包括Vec、HashMap、HashSet、BTreeMap和LinkedList。下面是使用我们讨论过的五种集合类型进行链式操作的一些示例：

1，Vec

let numbers = vec![1, 2, 3, 4, 5];

// Filter out odd numbers and square the remaining even numbers
let even_squares: Vec<_> = numbers
    .iter()
    .filter(|&n| n % 2 == 0)
    .map(|n| n * n)
    .collect();
println!("Even squares: {:?}", even_squares);

输出

Even squares: [4, 16]

在本例中，我们从Vec中过滤出奇数，对剩余的偶数进行平方，并将结果存储在一个新的Vec中。

2，HashMap

use std::collections::HashMap;

let mut scores = HashMap::new();
scores.insert("Alice", 90);
scores.insert("Bob", 80);
scores.insert("Carol", 70);
// Filter out entries with a score below 80 and create a new HashMap
let high_scores: HashMap<_, _> = scores
    .iter()
    .filter(|&(_, score)| *score >= 80)
    .map(|(name, score)| (name.to_string(), *score))
    .collect();
println!("High scores: {:?}", high_scores);

输出

High scores: {"Alice": 90, "Bob": 80}

在本例中，我们从HashMap中过滤出得分低于80的条目，并创建一个只包含高分的新HashMap。

3，HashSet

use std::collections::HashSet;

let set1: HashSet<_> = [1, 2, 3, 4, 5].iter().cloned().collect();
let set2: HashSet<_> = [4, 5, 6, 7, 8].iter().cloned().collect();
// Compute the intersection of two sets
let intersection: HashSet<_> = set1.intersection(&set2).cloned().collect();
println!("Intersection: {:?}", intersection);

输出

Intersection: {4, 5}

在本例中，我们计算两个HashSet的交集，并创建一个包含结果的新HashSet。

4，BTreeMap

use std::collections::BTreeMap;

let mut scores = BTreeMap::new();
scores.insert("Alice", 90);
scores.insert("Bob", 80);
scores.insert("Carol", 70);

// Map scores to grades and create a new BTreeMap
let grades: BTreeMap<_, _> = scores
    .iter()
    .map(|(name, score)| {
        let grade = if *score >= 90 {
            "A"
        } else if *score >= 80 {
            "B"
        } else {
            "C"
        };
        (name.to_string(), grade.to_string())
    })
    .collect();

println!("Grades: {:?}", grades);

输出

Grades: {"Alice": "A", "Bob": "B", "Carol": "C"}

在本例中，我们将分数映射到等级，并创建一个新的BTreeMap，其中姓名作为键，对应的等级作为值。

5，LinkedList

use std::collections::LinkedList;

let mut names = LinkedList::new();
names.push_back("Alice");
names.push_back("Bob");
names.push_back("Carol");
names.push_back("Dave");
// Create a new LinkedList containing only names starting with 'C'
let names_starting_with_c: LinkedList<_> = names
    .iter()
    .filter(|name| name.starts_with('C'))
    .cloned()
    .collect();
println!("Names starting with 'C':");
for name in names_starting_with_c {
    println!("{}", name);
}

输出

Names starting with 'C':
Carol

在这个例子中，我们从LinkedList中过滤掉不以'C'开头的名字，并创建一个新的LinkedList，只包含以'C'开头的名字。

类型转换

在Rust中，不同集合类型之间的转换是一种常见操作。collect()函数是Iterator trait提供的一个功能强大且通用的方法，它允许你将一种集合类型转换为另一种。下面是我们讨论过的五种集合类型之间的类型转换示例：

1，Vec to HashMap

let words = vec!["apple", "banana", "orange"];

// Create a HashMap where keys are words and values are their lengths
let word_lengths: HashMap<_, _> = words
    .iter()
    .map(|word| (word, word.len()))
    .collect();
println!("Word lengths: {:?}", word_lengths);

输出

Word lengths: {"apple": 5, "banana": 6, "orange": 6}

在本例中，我们将单词的Vec转换为HashMap，其中键是单词，值是单词的长度。

2，HashMap to Vec

use std::collections::HashMap;

let mut scores = HashMap::new();
scores.insert("Alice", 90);
scores.insert("Bob", 80);
scores.insert("Carol", 70);
// Create a Vec of names sorted by score
let sorted_names: Vec<_> = scores
    .iter()
    .sorted_by_key(|&(_, score)| score)
    .map(|(name, _)| name)
    .collect();
println!("Sorted names by score: {:?}", sorted_names);

输出

Sorted names by score: ["Carol", "Bob", "Alice"]

在本例中，我们将名称和分数的HashMap转换为按分数排序的名称的Vec。

3，HashSet to Vec

use std::collections::HashSet;

let set: HashSet<_> = [1, 2, 3, 4, 5].iter().cloned().collect();
// Create a Vec containing elements from the HashSet
let vec: Vec<_> = set.into_iter().collect();
println!("Vec from HashSet: {:?}", vec);

输出

Vec from HashSet: [1, 2, 3, 4, 5]

在本例中，我们将整数的HashSet转换为包含相同元素的Vec。

4，BTreeMap to Vec

use std::collections::BTreeMap;

let mut scores = BTreeMap::new();
scores.insert("Alice", 90);
scores.insert("Bob", 80);
scores.insert("Carol", 70);
// Create a Vec of names sorted by score
let sorted_names: Vec<_> = scores
    .iter()
    .map(|(name, _)| name)
    .collect();
println!("Sorted names by score: {:?}", sorted_names);

输出

Sorted names by score: ["Alice", "Bob", "Carol"]

在这个例子中，我们将名称和分数的BTreeMap转换为按分数排序的名称Vec。

5，LinkedList to Vec

use std::collections::LinkedList;

let mut names = LinkedList::new();
names.push_back("Alice");
names.push_back("Bob");
names.push_back("Carol");
// Create a Vec containing elements from the LinkedList
let vec: Vec<_> = names.into_iter().collect();
println!("Vec from LinkedList: {:?}", vec);

输出

Vec from LinkedList: ["Alice", "Bob", "Carol"]

在本例中，我们将名称的LinkedList转换为包含相同元素的Vec。

注意事项

1，从迭代器开始：要开始链接操作，首先需要一个迭代器。Rust中的大多数集合都有iter()、iter_mut()和into_iter()等方法，可以提供迭代器。

2，有效地使用闭包：许多迭代器方法，如map()、filter()和fold()，都接受闭包作为参数。使用闭包来应用转换、筛选元素或累积值。

3，惰性是你的朋友：像map()和filter()这样的迭代器方法是惰性的，这意味着它们直到你使用迭代器才执行(例如，通过调用collect()或使用for循环)。这允许你在不牺牲性能的情况下构建复杂的操作链。

4，收集结果：使用collect()方法将最终迭代器转换回集合类型。你可以显式地指定结果集合的类型，或者让Rust根据上下文推断类型。

5，来自其他trait的链式操作：Iterator方法可以与来自其他trait的方法组合，例如itertools库中的Iterator::sorted_by_key()，这使得操作链更加强大和富有表现力。

6，注意性能影响：虽然链式操作可以产生干净简洁的代码，但请记住，一些迭代器方法可能会影响性能。例如，使用collect()创建中间集合可能导致额外的分配，而链接许多flat_map()调用可能导致嵌套循环。

7，考虑对更复杂的转换使用链式操作：如果你发现应用需要按顺序进行多次转换，考虑使用链式操作让你的代码更具可读性和可维护性。通过链式操作，你可以将一系列转换表示为一个单一的、连贯的管道，从而更容易理解整个过程。

8，测试和基准测试你的链式操作：要确保链式操作按预期执行，请编写测试和基准测试来测量它们的正确性和性能。这将帮助你识别代码中潜在的瓶颈或低效操作。

9，利用Option和Result类型的强大功能：许多迭代器方法与Option和Result类型无缝配合，允许你在链式操作中优雅地处理错误和可选值。例如，你可以使用filter_map()在处理None值的同时过滤和转换元素，或者使用try_fold()在传播错误的同时积累值。

10，保持链式操作的可读性：虽然链式操作可以产生简洁的代码，但过长或过复杂的链式操作会变得难以阅读和理解。将长的链式操作分解为更小的、名称良好的中间变量或函数，以保持代码清晰和可维护。

总结

通过遵循这些技巧，你可以在Rust中利用链操作的强大功能来编写高效、有表现力且可读的代码。记住，在应用这些技术时，要始终考虑用例的特定需求，并在性能和可读性之间取得平衡。

本文由微信公众号coding到灯火阑珊原创，哈喽比特收录。
文章来源：https://mp.weixin.qq.com/s/9duEcev-sq9bPeukgZVFKA