Question

Go是否与Python的in关键字类似？我想检查一个值是否在列表中。

例如在Python中：

x = 'red'

if x in ['red', 'green', 'yellow', 'blue']:
    print "found"
else:
    print "not found"

在Go中，我想出了使用set惯用法，但我不认为它是理想的，因为我必须指定一个我不使用的int值。

x := "red"

valid := map[string]int{"red": 0, "green": 0,"yellow": 0, "blue": 0}

if _, ok := valid[x]; ok {
    fmt.Println("found")
} else {
    fmt.Println("not found")
}

我知道in关键字可能与泛型相关。有没有办法使用go generate或其他东西？

Answer 1

您可以使用map[string]bool作为一组。在测试并且某个键不在地图中时，会返回bool的零值false。

因此请使用有效值作为键并使用true作为值填充地图。如果测试的键值在地图中，则其存储的true值将是结果。如果测试的键值不在映射中，则返回值类型的零值，即false。

使用它，测试变得如此简单：

valid := map[string]bool{"red": true, "green": true, "yellow": true, "blue": true}

if valid[x] {
    fmt.Println("found")
} else {
    fmt.Println("not found")
}

在Go Playground上尝试（使用下面提到的变体）。

博客文章中提到了这一点：Go maps in action: Exploiting zero values

注意：

如果您有许多有效值，因为要存储在地图中的所有值都是true，使用切片列出有效值并使用for range循环可能会更紧凑初始化地图，如下所示：

for _, v := range []string{"red", "green", "yellow", "blue"} { valid[v] = true }

注意＃2：

如果您不想使用for range循环初始化，您仍然可以通过创建无类型（或bool - 类型）单字母const来优化它：

const t = true valid := map[string]bool{"red": t, "green": t, "yellow": t, "blue": t}

Answer 2

我认为另一个答案中的 map[string]bool 是一个不错的选择。另一种方法是 map[string]struct{}，它使用的内存略少：

package main

func main() {
   x, valid := "red", map[string]struct{}{
      "red": {}, "green": {}, "yellow": {}, "blue": {},
   }
   if _, ok := valid[x]; ok {
      println("found")
   } else {
      println("not found")
   }
}

你也可以把它包装成一个类型：

package main

type set map[string]struct{}

func newSet(slice []string) set {
   s := make(set)
   for _, each := range slice {
      s[each] = struct{}{}
   }
   return s
}

func (s set) has(v string) bool {
   _, ok := s[v]
   return ok
}

func main() {
   x := "red"
   if newSet([]string{"red", "green", "yellow", "blue"}).has(x) {
      println("found")
   } else {
      println("not found")
   }
}

检查值是否在列表中

2 个答案: