返回字符串的前n个字符

时间:2017-01-11 22:55:02

标签: go

将前n个字符作为字符串的子字符串返回的最佳方法是什么,当字符串中没有字符时,只返回字符串本身。

我可以做以下事情:

func firstN(s string, n int) string {
     if len(s) > n {
          return s[:n]
     }
     return s
}

但是有更清洁的方式吗?

BTW,在Scala中,我可以做s take n

3 个答案:

答案 0 :(得分:2)

除非您想使用unicode,否则您的代码很好:

fmt.Println(firstN("世界 Hello", 1)) // �

要使其与unicode一起使用,您可以通过以下方式修改该功能:

// allocation free version
func firstN(s string, n int) string {
    i := 0
    for j := range s {
        if i == n {
            return s[:j]
        }
        i++
    }
    return s
}
fmt.Println(firstN("世界 Hello", 1)) // 世

// you can also convert a string to a slice of runes, but it will require additional memory allocations
func firstN2(s string, n int) string {
    r := []rune(s)
    if len(r) > n {
        return string(r[:n])
    }
    return s
}
fmt.Println(firstN2("世界 Hello", 1)) // 世

答案 1 :(得分:0)

对我来说这看起来很干净。我会做同样的事情。

答案 2 :(得分:0)

  college := "ARMY INSTITUTE OF TECHNOLOGY PUNE"
  fmt.Println(college)
 
  name :=  str[0:4]
  fmt.Println(name)