本文探讨了在Go语言中使用 range 迭代切片时，如何获取切片元素的引用以进行修改。通过分析常见的错误用法，并提供优化的代码示例，阐述了直接通过索引访问切片元素和使用指针两种解决方案，帮助开发者更高效地操作切片数据。
在Go语言中，range 关键字提供了一种简洁的方式来迭代切片和数组。然而，当需要修改切片中的元素时，直接使用 range 可能会遇到问题，因为默认情况下 range 循环提供的是元素的副本而非引用。
理解 range 循环的行为
考虑以下代码：
package main
import "fmt"
type Account struct {
Balance int
}
type AccountList []Account
func main() {
accounts := AccountList{{Balance: 0}, {Balance: 0}, {Balance: 0}}
// 错误示例：修改的是副本
for _, a := range accounts {
a.Balance = 100
}
fmt.Println("错误示例结果:", accounts) // 输出：[{0} {0} {0}]
// 正确示例：通过索引修改
for i := range accounts {
accounts[i].Balance = 100
}
fmt.Println("正确示例结果:", accounts) // 输出：[{100} {100} {100}]
}在第一个循环中，变量 a 接收的是 accounts 切片中元素的副本。因此，对 a.Balance 的修改不会影响原始切片中的元素。 第二个循环通过索引 i 直接访问切片 accounts 中的元素，并修改其 Balance 字段，从而实现了预期的效果。
立即学习“go语言免费学习笔记（深入）”；
使用索引获取引用
虽然通过索引访问切片元素可以解决问题，但每次循环都需要进行一次索引查找。如果需要在循环中对同一个元素进行多次操作，可以先获取元素的指针，然后再进行操作，以提高效率。
package main
import "fmt"
type Account struct {
Balance int
}
type AccountList []Account
func main() {
accounts := AccountList{{Balance: 0}, {Balance: 0}, {Balance: 0}}
// 使用索引获取指针
for i := range accounts {
a := &accounts[i]
a.Balance = 100
// 对 account A 进行更多操作...
}
fmt.Println("使用索引获取指针结果:", accounts) // 输出：[{100} {100} {100}]
}在这个示例中，a := &accounts[i] 获取了 accounts[i] 的指针，后续对 a.Balance 的修改实际上是直接修改了切片中的元素。
使用指针切片
另一种解决方案是直接使用指针切片。也就是说，切片中的每个元素都是指向 Account 结构的指针。
package main
import "fmt"
type Account struct {
Balance int
}
type AccountList []*Account
func main() {
accounts := AccountList{&Account{Balance: 0}, &Account{Balance: 0}, &Account{Balance: 0}}
// 使用指针切片
for _, a := range accounts {
a.Balance = 100
}
fmt.Println("使用指针切片结果:")
for _, a := range accounts {
fmt.Printf("{%d} ", a.Balance) // 输出：{100} {100} {100}
}
fmt.Println()
}在这个示例中，accounts 是一个 *Account 类型的切片。range 循环中的 a 变量直接指向切片中的 Account 结构，因此对 a.Balance 的修改会直接影响原始数据。
总结
在Go语言中使用 range 迭代切片并修改元素时，需要注意 range 循环提供的是元素的副本。为了修改原始切片中的元素，可以使用以下两种方法：
- 通过索引访问切片元素：for i := range accounts { accounts[i].Balance = 100 }
- 使用指针切片：AccountList []*Account
选择哪种方法取决于具体的需求和场景。如果需要在循环中对同一个元素进行多次操作，建议使用索引获取指针，以提高效率。如果切片中的元素本身就是指针类型，那么可以直接使用 range 循环进行修改。
