collections - 构建字符计数 HashMap 的惯用 Rust 方法是什么？

Question

这个问题在这里已经有了答案:





How can I convert a collection of values into a HashMap that counts them?

(2 个回答)


1年前关闭。




我想计算字符串中每个字母的出现次数。目标是建立一个 HashMap<char,i32>其中键是字符串中的所有字符，值是出现次数。
假设我正在循环 char来自字符串或输入文件的值。对于每个 char ，如果还没有遇到，需要添加到HashMap作为一个值为 1 的新键，但如果它以前见过，我需要增加该值。
这是有效的代码。请耐心等待，我对 Rust 很陌生:

use std::collections::HashMap;

fn main() {
    let mut letter_counts: HashMap<char,i32> = HashMap::new();

    let input_string = "Hello, world!";
    let char_vec: Vec<char> = input_string.to_lowercase().chars().collect();
    for c in char_vec {
        if let Some(x) = letter_counts.get_mut(&c) {
            *x = *x + 1;
        } else {
            letter_counts.insert(c,1);
        }
    }
    println!("{:?}",letter_counts);
}

我想知道的是，在 Rust 中是否有一种惯用的方法来做到这一点？通过惯用语，我的意思是有一个标准库类型(如 Python 的 defaultdict )，或 HashMap 上的方法(如 Java 的 HashMap.computeIfAbsent )，这将使这比手动编码更简单、更清晰和/或更不容易出错我做过的算法？

Answer 1

如果您使用 Entry，这对于您想要做的事情可能会更容易一些，并且可能更惯用一些。界面:

use std::collections::HashMap;

fn main() {
    let mut letter_counts: HashMap<char,i32> = HashMap::new();

    let input_string = "Hello, world!";
    let char_vec: Vec<char> = input_string.to_lowercase().chars().collect();
    for c in char_vec {
        *letter_counts.entry(c).or_insert(0) += 1;
    }
    println!("{:?}",letter_counts);
}

如果条目不存在，您可以创建该条目并同时对其进行修改。
如果你想要一些更实用的东西，你可以这样做:

use std::collections::HashMap;

fn main() {
    let input_string = "Hello, world!";
    let letter_counts: HashMap<char, i32> =
        input_string
            .to_lowercase()
            .chars()
            .fold(HashMap::new(), |mut map, c| {
                *map.entry(c).or_insert(0) += 1;
                map
            });
    println!("{:?}", letter_counts);
}

这使用折叠来累积项目。
如果您正在寻找一个计算项目频率的标准库函数，那么没有。函数式方法足够优雅，我个人并不认为这是一个错误，而且在这种情况下，我通常会使用这种方法来处理大多数正常、惯用的 Rust 代码。使用迭代器在 Rust 中很常见。
正如其他人所提到的，对于某些情况，肯定有更专业的替代方法。