sunty
2024-09-03 b0f374409775bd30ed8f0dc8d47d015d3edc0abb
data/prepare.go
@@ -2,6 +2,7 @@
import (
   "fmt"
   "regexp"
   "ruleModelEngine/db"
   "time"
)
@@ -80,13 +81,35 @@
   // 找到出现次数最多的出行地址
   maxOutCount := 0
   maxTotalCount := 0
   var frequentAddress string
   floor1AddressCount := 0
   floor1Address := ""
   for address, count := range outAddressCounts {
      maxTotalCount = maxTotalCount + count
      re, _ := regexp.Compile(`\b1F\b`)
      if re.MatchString(address) {
         if count > floor1AddressCount {
            floor1Address = address
            floor1AddressCount = count
            continue
         }
      }
      if count > maxOutCount {
         maxOutCount = count
         frequentAddress = address
      }
   }
   if frequentAddress == "" && floor1Address != "" {
      frequentAddress = floor1Address
   }
   if frequentAddress != "" && floor1Address != "" {
      if (float64(floor1AddressCount)/float64(maxTotalCount))*100 > 90 {
         frequentAddress = floor1Address
      }
   }
   if frequentAddress == "" {
      maxInCount := 0
      for address, count := range inAddressCounts {
@@ -96,6 +119,9 @@
         }
      }
   }
   //fmt.Println("in address: ", inAddressCounts)
   //fmt.Println("out address: ", outAddressCounts)
   //fmt.Println("final address: ", frequentAddress)
   return frequentAddress
}