Go语言中切片迭代与元素修改的正确姿势

22次阅读

Go语言中切片迭代与元素修改的正确姿势

go语言中,使用`for…range`循环遍历切片时,理解其迭代行为对于正确访问和修改元素至关重要。本文将深入探讨`for…range`在单变量和双变量模式下的行为差异,特别是当尝试修改切片元素时可能遇到的常见陷阱,并提供通过索引进行修改的正确方法,以避免“undefined attributes”等编译错误

go语言中for…range切片迭代机制

for…range是Go语言中遍历切片、数组、字符串、映射和通道的强大工具。然而,对于切片,其行为在不同使用方式下有所区别,尤其是在尝试修改切片元素时,需要特别注意。

1. 单变量模式:获取索引

当for…range循环只使用一个变量时,该变量将接收当前迭代的索引

例如,在以下代码片段中:

func (g *graph) addnodes() {     g.nodes = make([]node, g.nodesnr)     for n := range g.nodes { // 这里的 n 是索引,类型为 int         n.value = 2         // 错误:int 类型没有 value 字段         n.neigbours = nil   // 错误:int 类型没有 neigbours 字段         return     } }

这里的n实际上是切片g.nodes中的元素的索引,其类型为int。因此,尝试访问n.value或n.neigbours会导致编译错误,因为int类型并没有这些字段。这是初学者常犯的错误,误以为n代表了切片中的元素本身。

立即学习go语言免费学习笔记(深入)”;

2. 双变量模式:获取索引和值(副本)

当for…range循环使用两个变量时,第一个变量接收索引,第二个变量接收当前迭代的。需要注意的是,这个值是切片元素的副本

for index, value := range mySlice {     // index 是当前元素的索引     // value 是当前元素的副本 }

如果只关心值而不关心索引,可以使用下划线_来忽略索引:

for _, n := range g.nodes { // 这里的 n 是 node 结构体的副本     // n.value = 2 // 此时 n 是 node 类型,可以访问 value 字段     // n.neigbours = nil     // 注意:修改 n 的字段不会影响原始切片中的元素,因为 n 是副本 }

在这种模式下,n确实是node类型,可以访问其字段。然而,由于n是切片元素的副本,对n的任何修改都不会反映到原始切片g.nodes中的对应元素上。如果node结构体中包含引用类型(如切片、映射、指针等),并通过n修改这些引用类型指向的数据,那么这些修改会影响原始数据,但直接修改n的非引用类型字段(如value int)则无效。

Go语言中切片迭代与元素修改的正确姿势

图改改

在线修改图片文字

Go语言中切片迭代与元素修改的正确姿势455

查看详情 Go语言中切片迭代与元素修改的正确姿势

正确修改切片元素的方法

要正确修改切片中的元素,必须通过元素的索引来访问和修改原始切片。

func (g *graph) addNodes() {     g.nodes = make([]node, g.nodesnr)     // 遍历索引,通过索引直接访问并修改切片中的元素     for i := range g.nodes { // i 是索引,类型为 int         g.nodes[i].value = 2         // 正确:通过索引访问原始元素并修改         g.nodes[i].neigbours = nil   // 正确:通过索引访问原始元素并修改         // 注意:这里的 return 语句会导致循环在第一次迭代后立即退出,         // 仅初始化了 g.nodes 中的第一个元素。如果需要初始化所有元素,应移除 return。     } }

在这个修正后的addNodes函数中,我们使用for i := range g.nodes来获取每个元素的索引i。然后,通过g.nodes[i]直接访问切片中的原始node结构体,并对其value和neigbours字段进行赋值。这样,修改将直接作用于切片中的实际元素。

示例代码

以下是修正后的addNodes函数,展示了如何正确地初始化切片中的所有node元素:

package main  import (     "fmt" )  type node struct {     value     int     neigbours []int }  type edge struct {     source int     sink   int }  type graph struct {     nodesnr, edgesnr int     nodes            []node     edges            chan edge }  func main() {     g := randomGraph()     // 打印所有节点的 value 以验证初始化     for i := 0; i < g.nodesnr; i++ {         fmt.Printf("Node %d value: %dn", i, g.nodes[i].value)     } }  func input(tname string) (number int) {     fmt.Println("input a number of " + tname)     fmt.Scan(&number)     return }  func randomGraph() (g graph) {     g = graph{nodesnr: input("nodes"), edgesnr: input("edges")}     g.addNodes() // 调用修正后的 addNodes     return }  func (g *graph) addNodes() {     g.nodes = make([]node, g.nodesnr)     // 正确地通过索引初始化切片中的所有元素     for i := range g.nodes {         g.nodes[i].value = i + 1 // 为每个节点设置一个不同的值,例如从1开始         g.nodes[i].neigbours = []int{} // 初始化为空切片,而不是 nil     } }  // 其他函数保持不变或根据需要进行修正 func (g *graph) addEdges() {     g.edges = make(chan edge)     // ... }  func (g *graph) edgeCheck(ep *edge) string {     // ...     return "empty" }  func (g *graph) neigbourCheck(neigbours []node, node int) bool {     // ...     return false }  func (g *graph) addEdge() {     // ... }

注意事项:

  1. for i := range slice 获取的是索引。
  2. for _, val := range slice 获取的是值的副本。 对val的修改不会影响原始切片元素,除非val本身是引用类型且你修改了它指向的数据。
  3. 要修改切片元素,请务必通过索引slice[i]进行操作。
  4. 在上述addNodes函数中,原始代码的return语句在循环内部,会导致循环只执行一次。如果目的是初始化所有元素,应移除return语句。

总结

理解for…range循环在Go语言中处理切片的方式是编写高效且无错误代码的关键。当需要修改切片中的元素时,始终通过索引访问它们是最佳实践。通过区分获取索引、获取值副本以及通过索引修改原始元素的机制,可以避免常见的编译错误和逻辑错误,确保程序行为符合预期。

text=ZqhQzanResources