看代码吧~
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
func remove(slice []interface{}, elem interface{}) []interface{}{
if len(slice) == 0 {
return slice
}
for i, v := range slice {
if v == elem {
slice = append(slice[:i], slice[i+1:]...)
return remove(slice,elem)
break
}
}
return slice
}
func removeZero(slice []interface{}) []interface{}{
if len(slice) == 0 {
return slice
}
for i, v := range slice {
if ifZero(v) {
slice = append(slice[:i], slice[i+1:]...)
return removeZero(slice)
break
}
}
return slice
}
|
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
//判断一个值是否为零值，只支持string,float,int,time 以及其各自的指针，"%"和"%%"也属于零值范畴，场景是like语句
func IfZero(arg interface{}) bool {
if arg == nil {
return true
}
switch v := arg.(type) {
case int, int32, int16, int64:
if v == 0 {
return true
}
case float32:
r:=float64(v)
return math.Abs(r-0)<0.0000001
case float64:
return math.Abs(v-0)<0.0000001
case string:
if v == "" || v == "%%" || v == "%" {
return true
}
case *string, *int, *int64, *int32, *int16, *int8, *float32, *float64, *time.Time:
if v == nil {
return true
}
case time.Time:
return v.IsZero()
default:
return false
}
return false
}
|
补充：golang删除slice中特定条件的元素，优化版
写了两种对一个slice中删除特定元素的方法，并做了性能对比，在这里记录一下。
假设我们的切片有0和1，我们要删除所有的0，此处有三种方法：
第一种方法：
|
1
2
3
4
5
6
7
8
9
|
func DeleteSlice(a []int) []int{
for i := 0; i < len(a); i++ {
if a[i] == 0 {
a = append(a[:i], a[i+1:]...)
i--
}
}
return a
}
|
解释：这里利用常见的方法对slice中的元素进行删除，注意删除时，后面的元素前移，i应该后移一位。
第二种方法：
|
1
2
3
4
5
6
7
8
9
|
func DeleteSlice1(a []int) []int {
ret := make([]int, 0, len(a))
for _, val := range a {
if val == 1 {
ret = append(ret, val)
}
}
return ret
}
|
解释：这种方法最容易理解，重新使用一个slice，将不合理的过滤掉。缺点是需要开辟另一个slice的空间，优点是容易理解，而且不对原来的slice进行操作。
第三种方法：
|
1
2
3
4
5
6
7
8
9
10
|
func DeleteSlice2(a []int) []int{
j := 0
for _, val := range a {
if val == 1 {
a[j] = val
j++
}
}
return a[:j]
}
|
解释：这里利用一个index，记录应该下一个有效元素应该在的位置，遍历所有元素，当遇到有效元素，index加一，否则不加，最终index的位置就是所有有效元素的下一个位置。最后做一个截取就行了。这种方法会对原来的slice进行修改。
这里对三种方法做了性能测试，测试代码如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
package main
import (
"testing"
)
func handle(data []int) {
return
}
const N = 100
func getSlice()[]int {
a := []int{}
for i := 0; i < N; i++ {
if i % 2 == 0 {
a = append(a, 0)
} else {
a = append(a, 1)
}
}
return a
}
func BenchmarkDeleteSlice(b *testing.B) {
for i := 0; i < b.N; i++ {
data := DeleteSlice(getSlice())
handle(data)
}
}
func BenchmarkDeleteSlice1(b *testing.B) {
for i := 0; i < b.N; i++ {
data := DeleteSlice1(getSlice())
handle(data)
}
}
func BenchmarkDeleteSlice2(b *testing.B) {
for i := 0; i < b.N; i++ {
data := DeleteSlice2(getSlice())
handle(data)
}
}
|
测试结果如下（slice大小为100）：
加大slice大小进行测试（slice大小为10000）：
继续加大（slice大小为100000）
slice大小为10^6:
可以看出：
第一种方法在slice大小比较小时，比第2、3种方法慢一倍左右。但是slice大小变大时，性能显著下降。
第2种方法和第3种方法差距基本处于同一量级，但是第3种方法稍快一些。但是当slice大小增加到10^6级别时，第三种方法的优势就显现出来。
以上为个人经验，希望能给大家一个参考，也希望大家多多支持服务器之家。如有错误或未考虑完全的地方，望不吝赐教。
原文链接：https://blog.csdn.net/fwhezfwhez/article/details/79931415