如何更新可变 HashMap 中的值？

Question

这是我正在尝试做的事情：

use std::collections::HashMap;

fn main() {
    let mut my_map = HashMap::new();
    my_map.insert("a", 1);
    my_map.insert("b", 3);

    my_map["a"] += 10;
    // I expect my_map becomes {"b": 3, "a": 11}
}

但这会引发错误：

生锈 2015

error[E0594]: cannot assign to immutable indexed content
 --> src/main.rs:8:5
  |
8 |     my_map["a"] += 10;
  |     ^^^^^^^^^^^^^^^^^ cannot borrow as mutable
  |
  = help: trait `IndexMut` is required to modify indexed content, but it is not implemented for `std::collections::HashMap<&str, i32>`

生锈 2018

error[E0594]: cannot assign to data in a `&` reference
 --> src/main.rs:8:5
  |
8 |     my_map["a"] += 10;
  |     ^^^^^^^^^^^^^^^^^ cannot assign

我真的不明白那是什么意思，因为我把 HashMap 变成了可变的。当我尝试更新 vector 中的元素时，我得到了预期的结果：

let mut my_vec = vec![1, 2, 3];

my_vec[0] += 10;
println! {"{:?}", my_vec};
// [11, 2, 3]

HashMap 我收到上述错误有何不同？有没有办法更新值？

Answer 1

不可变索引和可变索引分别由两个不同的特征提供：Index and IndexMut。

目前，HashMap does not implement IndexMut, while Vec does。

The commit that removed HashMap's IndexMut implementation 状态：

This commit removes the IndexMut impls on HashMap and BTreeMap, in order to future-proof the API against the eventual inclusion of an IndexSet trait.

据我了解，假设的 IndexSet 特征允许您为 HashMap 分配全新的值，而不仅仅是读取或改变现有条目：

let mut map = HashMap::new();
map["key"] = "value";

目前，您可以使用 get_mut:

*my_map.get_mut("a").unwrap() += 10;

或者 entry API:

*my_map.entry("a").or_insert(42) += 10;

Answer 2

正在考虑：

let mut m = std::collections::HashMap::new();
m.insert("a", 1);
m.insert("b", 3);
let k = "c";

如果密钥已经存在：

    m.insert(k, 10 + m[k] );

如果密钥不存在：

您可以更新键值：

    m.insert(k, 10 + if m.contains_key(k) { m[k] } else { 0 });

或者先插入一个键，如果它不存在的话：

    m.entry(k).or_insert(0);
    m.insert(k, 200 + m[k]);

或者更新一个key，防止key可能没有被设置：

    *m.entry(k).or_insert(0) += 3000;

最后打印值：

    println!("{}", m[k]); // 3210

参见：
https://doc.rust-lang.org/std/collections/struct.HashMap.html

Answer 3

我将分享我自己的答案，因为我遇到了这个问题，但我正在使用 Structs，所以在我的情况下，这种方式有点棘手

use std::collections::HashMap;

#[derive(Debug)]
struct ExampleStruct {
    pub field1: usize,
    pub field2: f64,
}

fn main() {
    let mut example_map = HashMap::new();
    &example_map.insert(1usize, ExampleStruct { field1: 50, field2: 184.0});
    &example_map.insert(6usize, ExampleStruct { field1: 60, field2: 486.0});

    //First Try
    (*example_map.get_mut(&1).unwrap()).field1 += 55; //50+55=105
    (*example_map.get_mut(&6).unwrap()).field1 -= 25; //60-25=35

    //Spliting lines
    let op_elem = example_map.get_mut(&6);
    let elem = op_elem.unwrap();
    (*elem).field2 = 200.0;

    let op_ok_elem = example_map.get_mut(&1);
    let elem = op_ok_elem.unwrap_or_else(|| panic!("This msg should not appear"));
    (*elem).field2 = 777.0;

    println!("Map at this point: {:?}", example_map);
    let op_err_elem = example_map.get_mut(&8);
    let _elem = op_err_elem.unwrap_or_else(|| panic!("Be careful, check you key"));

    println!("{:?}", example_map);
}

您可以在 Rust Playground

上玩这个

如何更新可变 HashMap 中的值？

How can I update a value in a mutable HashMap?

hashmap

rust