循环插入hashmap

Question

我正在打开一个 CSV 文件并使用 BufReader 读取它并将每一行拆分为一个向量。然后我尝试使用特定列作为键在HashMap 中插入或更新计数。

let mut map: HashMap<&str, i32> = HashMap::new();

let reader = BufReader::new(input_file);
for line in reader.lines() {
    let s = line.unwrap().to_string();
    let tokens: Vec<&str> = s.split(&d).collect(); // <-- `s` does not live long enough
    if tokens.len() > c {
        println!("{}", tokens[c]);
        let count = map.entry(tokens[c].to_string()).or_insert(0);
        *count += 1;
    }
}

编译器好心地告诉我s 是短暂的。 Storing from inside a loop a borrowed value to container in outer scope? 建议“拥有”字符串，所以我尝试更改

let count = map.entry(tokens[c]).or_insert(0);

到

let count = map.entry(tokens[c].to_string()).or_insert(0);

但我得到了错误

expected `&str`, found struct `std::string::String`
help: consider borrowing here: `&tokens[c].to_string()`

当我在 & 前面加上 (&) 时，错误是

creates a temporary which is freed while still in use
note: consider using a `let` binding to create a longer lived

我对借用的 Rust 知识有一些不足。如何让 hashmap 拥有作为键传递的字符串？

Answer 1

最简单的方法是让您的地图拥有密钥。这意味着您必须将其类型从HasMap<&str, i32>（借用密钥）更改为HashMap<String, i32>。此时您可以调用to_string 将您的令牌转换为拥有的字符串：

let mut map: HashMap<String, i32> = HashMap::new();

let reader = BufReader::new(input_file);
for line in reader.lines() {
    let s = line.unwrap().to_string();
    let tokens:Vec<&str> = s.split(&d).collect();
    if tokens.len() > c {
        println!("{}", tokens[c]);
        let count = map.entry(tokens[c].to_string()).or_insert(0);
        *count += 1;
    }
}

但是请注意，这意味着 tokens[c] 将被复制，即使它已经存在于地图中。您可以先尝试使用get_mut 修改计数器来避免额外的重复，但这需要在缺少键时进行两次查找：

let mut map: HashMap<String, i32> = HashMap::new();

let reader = BufReader::new(input_file);
for line in reader.lines() {
    let s = line.unwrap().to_string();
    let tokens:Vec<&str> = s.split(&d).collect();
    if tokens.len() > c {
        println!("{}", tokens[c]);
        if let Some (count) = map.get_mut (tokens[c]) {
            *count += 1;
        } else {
            map.insert (tokens[c].to_string(), 1);
        }
    }
}

我不知道只有在没有先前条目时才复制密钥但仍进行单次查找的解决方案。