使用折叠迭代器方法在 Rust 中创建 HashMap

Question

我正在尝试使用 Entry 和 Rust 中的迭代器从单词数组中创建字数 HashMap。当我尝试如下时，

use std::collections::HashMap;
fn main () {
    let corpus = ["foo", "bar", "bar", "grok", "blah", "foo", "bar"];
    let mut word_count = corpus.iter().fold(HashMap::new(), |mut acc, word| *acc.entry(word).or_insert(0) += 1);
    println!("{:?}", word_count);
}

我收到错误：预期结构 HashMap，找到 ()，类型不匹配。

我可以通过创建一个新范围并将累加器显式返回为：

use std::collections::HashMap;
fn main () {
    let corpus = ["foo", "bar", "bar", "grok", "blah", "foo", "bar"];
    let mut word_count = corpus.iter().fold(HashMap::new(), |mut acc, word|{
        *acc.entry(word).or_insert(0) += 1;
        acc
    });
    println!("{:?}", word_count);
}

我得到了预期的{"foo": 2, "bar": 3, "blah": 1, "grok": 1}。

通常fold 返回累加器的最后状态，所以我期待第一种方法可以工作，但不清楚为什么它不工作。非常感谢您对替代方法（使用迭代器）进行一些澄清和任何建议。

Answer 1

就像@PitaJ 的评论中解释的那样，*acc.entry(word).or_insert(0) += 1 的类型为()，而编译器期望fold() 的回调每次都返回下一个状态（在其他语言中有时称为reduce()，例如 JavaScript; 在 Rust 中；fold() 允许您指定起始值，而 reduce() 从迭代器中的第一项获取它。

因此，我觉得循环更适合用例，因为您不需要返回新状态而是更新地图：

let mut word_count = HashMap::new();
for word in &corpus {
    *word_count.entry(word).or_insert(0) += 1;
}

Playground.

但是，如果您愿意，itertools crate 中有一个方法可以做到这一点：counts():

let word_count = corpus.iter().counts();

Playground.