Go 修改map slice array元素值操作

Posted on by

在“range” 語句中生成的數據的值其實是集合元素的拷貝。它們不是原有元素的引用。

這就意味著更新這些值將不會修改原來的數據。

我們來直接看段示例：

package main
import "fmt"
func main() {
 data := []int{1, 2, 3}
 for _, v := range data {
  v *= 10 //原始元素未更改
 }
 fmt.Println("data:", data) //輸出 data: [1 2 3]
}

如果我們需要更新原有集合中的數據，使用索引操作符來獲得數據即可：

package main
import "fmt"
func main() {
 data := []int{1, 2, 3}
 for i, _ := range data {
  data[i] *= 10
 }
 fmt.Println("data:", data) //輸出 data: [10 20 30]
}

好，重點來瞭!重點來瞭!重點來瞭!重要的話說三遍，大部分博友們可能會踩坑.

這裡我提前總結下：

多個slice可以引用同一個數據。比如，當你從一個已有的slice創建一個新的slice時(比如通過索引截取)，這就會發生。

如果你的應用功能需要這種行為，那麼你將需要留意下slice的”坑”。

在某些情況下，在一個slice中添加新的數據，在原有數組無法保持更多新的數據時，將導致分配一個新的數組。

而其他的slice還指向老的數組（或者是老的數據）。

package main
import "fmt"
func main() {
 s1 := []int{1, 2, 3}
 fmt.Println(len(s1), cap(s1), s1) //輸出 3 3 [1 2 3]
 s2 := s1[1:] //索引從第二個元素截取開始
 fmt.Println(len(s2), cap(s2), s2) //輸出 2 2 [2 3]
 for i := range s2 {
  s2[i] += 20
 }
 //仍然引用同一數組
 fmt.Println(s1) //s1 在s2修改瞭後面2個元素，所以s1也是更新瞭。輸出 [1 22 23]
 fmt.Println(s2) //輸出 [22 23]
 s2 = append(s2, 4) // 註意s2的容量是2，追加新元素後將導致分配一個新的數組 [22 23 4]
 for i := range s2 {
  s2[i] += 10
 }
 //s1 仍然是更新後的歷史老數據
 fmt.Println(s1) //輸出 [1 22 23]
 fmt.Println(s2) //輸出 [32 33 14]
}

所以，大傢在使用中特別註意。容量不足，追加新元素不影響歷史數據。因為重新分配瞭變量瞭。

另外，繼續聊下高級一點滴技巧：

使用指針接收方法的值

隻要值是可取址的，那在這個值上調用指針接收方法是沒問題的。

然而並不是所有的變量是可取址的。Map的元素就不是。通過interface引用的變量也不是。我們接著看下面一段代碼：

package main
import "fmt"
type user struct {
 name string
}
func (p *user) print() {
 fmt.Println("排名:", p.name)
}
type printer interface {
 print()
}
func main() {
 u := user{"喬峰"}
 u.print()     // 輸出 排名: 喬峰
 var in printer = user{"鳩摩智"} //error
 in.print()
 m := map[string]user{"one": user{"風清揚"}}
 m["one"].print() //error
}

輸出：

cannot use user literal (type user) as type printer in assignment:
  user does not implement printer (print method has pointer receiver)
cannot call pointer method on m["one"]
cannot take the address of m["one"]

大致意思是：不能在賦值中使用數據文本（類型數據）作為類型指針，user未執行指針調用（指針方法具有指針接收器)，

無法對m[“one”]調用指針方法，不能取m的地址[“one”]。

上面我們看到有一個struct值的map，我們無法更新單個的struct值。比如錯誤的代碼：

package main
type user struct {
 name string
}
func main() {
 m := map[string]user{"one": {"喬峰"}}
 m["one"].name = "風清揚" //輸出 cannot assign to struct field m["one"].name in map
}

錯誤意思是：在map中，無法分配給結構字段m[“one”].name。這個操作無效是因為map元素是無法取址的。

上面我們提到:slice元素是可以取地址滴：

package main
import "fmt"
type user struct {
 name string
}
func main() {
 one := user{"喬峰"}
 u := []user{one}
 u[0].name = "風清揚" //ok
 fmt.Println(u) //輸出: [{風清揚}]
}

當然我們還有更好的解決辦法：

第一個有效的方法是使用一個臨時變量：

package main
import "fmt"
type user struct {
 name string
}
func main() {
 m := map[string]user{"one": {"喬峰"}}
 u := m["one"] //使用臨時變量
 u.name = "風清揚"
 m["one"] = u
 fmt.Printf("%v\n", m) //輸出: map[one:{風清揚}]
}

另一個有效的方法是使用指針的map：

package main
import "fmt"
type user struct {
 name string
}
func main() {
 m := map[string]*user{"one": {"喬峰"}}
 m["one"].name = "風清揚" //ok
 fmt.Println(m["one"]) //輸出: &{風清揚}
}

說到這裡，順便再提一下。繼續看下面一段代碼：

package main
import "fmt"
type user struct {
 name string
}
func main() {
 m := map[string]*user{"one": {"喬峰"}}
 m["two"].name = "鳩摩智" //新增自定義鍵名值
 fmt.Println(m["two"]) //error
}

輸出：

panic: runtime error: invalid memory address or nil pointer dereference

無效的內存地址或取消引用空指針？原因在於Go無法動態給結構體添加字段，我們可以間接使用make(map[string]interface{})實現。

好吧，就說這麼多瞭，有不足之處歡迎廣大博友留言指正。。。。。。。

補充：golang 中map 和slice 索引速度比較

主文件

package main
var max = 100
var Slice = make([]int, max+10)
var Map = make(map[int]int)
func init() {
 for i := 0; i < max; i++ {
 Slice[i] = i
 Map[i] = i
 }
}
// 查找算法可以優化，本文對於常用無序查找做比較
func SearchSlice(i int) int {
 for _, v := range Slice {
 if v == i {
 return v
 }
 }
 return -1
}
func SearchMap(i int) int {
 return Map[i]
}

測試文件

package main
import "testing"
func BenchmarkSearchMap(b *testing.B) {
 for i := 0; i < b.N; i++ {
 _ = SearchMap(i % max)
 }
}
func BenchmarkSearchSlice(b *testing.B) {
 for i := 0; i < b.N; i++ {
 _ = SearchSlice(i % max)
 }
}
func BenchmarkSlice(b *testing.B) {
 for i := 0; i < b.N; i++ {
 _ = Slice[i%max]
 }
}

測試結果

max = 100

BenchmarkSearchMap-16   94148293    12.7 ns/op    0 B/op   0 allocs/op
BenchmarkSearchSlice-16   49473447    23.6 ns/op    0 B/op   0 allocs/op
BenchmarkSlice-16    187461336    6.46 ns/op   0 B/op   0 allocs/op

max = 10000

BenchmarkSearchMap-16   43147364    27.6 ns/op    0 B/op   0 allocs/op
BenchmarkSearchSlice-16   968623    1159 ns/op    0 B/op   0 allocs/op
BenchmarkSlice-16    187649472    6.42 ns/op   0 B/op   0 allocs/op

Max = 1000000

BenchmarkSearchMap-16     15015690    90.1 ns/op    0 B/op   0 allocs/op
BenchmarkSearchSlice-16     441436   104242 ns/op    0 B/op   0 allocs/op
BenchmarkSlice-16      182620702    6.58 ns/op   0 B/op   0 allocs/op

在一些特定優化條件下，可以嘗試用slice，效果會比map好，比如把10 6級的查找優化成3級102查找, 對於一些結構體，可以根據某些特征分類或預先根據特征值排序。

以上為個人經驗，希望能給大傢一個參考，也希望大傢多多支持WalkonNet。如有錯誤或未考慮完全的地方，望不吝賜教。

推薦閱讀：