在go语言中计算相对强弱指数（RSI）示例代码

Go  /  管理员 发布于 1年前   612

下面是一个如何使用 Binance.com 和 Golang 的数据计算相对强弱指数（RSI）的简单示例。

如果您从 Binance 以外的交易所获取数据，只需稍作修改，本示例也能正常运行。

您所需要的只是一个包含收盘价值的切片。

更新：

请使用 "github.com/markcheno/go-talib" 中的 RSI 函数，它与 TradingView.com 的结果一致。

示例代码：

 package main
 
 import (
     "context"
     "fmt"
     "math"
     "strconv"
     "time"
     "github.com/adshao/go-binance"
 )
 
 var (
 //Binance
 binanceAPIKey    = "<your binance api key>"
 binanceSecretKey = "<your binance secret key>"
 )
 
 func humanTimeToUnixNanoTime(input time.Time) int64 {
 return int64(time.Nanosecond) * input.UnixNano() / int64(time.Millisecond)
 }
 
 func main() {
     client := binance.NewClient(binanceAPIKey, binanceSecretKey)
     targetSymbol := "BNBUSDT"
     
     // reference https://golang.org/pkg/time/#Date
     // func Date(year int, month Month, day, hour, min, sec, nsec int, loc *Location) Time
     
     // 注意：Binance 只接受 unix nano 时间戳，不接受 unix 时间戳。
     // 例如
     // 1527616800 <----- 将返回空结果或错误的日期范围
     // 1527616899999 <--- ok
     
     //startTimeHumanReadable := time.Date(2018, time.October, 10, 0, 0, 0, 0, time.UTC)
     //start := humanTimeToUnixNanoTime(startTimeHumanReadable)
     
     //endTimeHumanReadable := time.Date(2018, time.October, 24, 0, 0, 0, 0, time.UTC)
     //end := humanTimeToUnixNanoTime(endTimeHumanReadable)
     targetInterval := "1d" /// 区分大小写，1D != 1d 或 30M != 30m
     data, err := client.NewKlinesService().Symbol(targetSymbol).Interval(targetInterval).Do(context.Background())
     if err != nil {
     fmt.Println(err)
         return
     }
     totalGain := 0.0
     totalLoss := 0.0
     
     for i := 1; i < len(data); i++ {
         previous := data[i].Close
         current := data[i-1].Close
         
         // 将字符串转换为 float64
         previousClose, _ := strconv.ParseFloat(previous, 64)
         currentClose, _ := strconv.ParseFloat(current, 64)
         difference := currentClose - previousClose
         
         if difference >= 0 {
             totalGain += difference
         } else {
             totalLoss -= difference
         }
     }
     
     rs := totalGain / math.Abs(totalLoss)
     rsi := 100 - 100/(1+rs)
     fmt.Println("RSI for "+targetSymbol+" : ", rsi)
 }

输出示例：

Thu Oct 25 11:35:33 +08 2018
BNBUSDT 的 RSI : 47.84161851469368

注意：

请使用 "github.com/markcheno/go-talib" 中的 RSI 函数，它与 TradingView.com 的结果一致。

我试着比较结果是否与 TradingView.com (TV) 计算的 RSI 值相匹配，结果发现 BTCUSDT 和 BNBUSDT 的 RSI 值与本代码示例计算的值非常接近。

我注意到，在 TV 的 RSI 行中，有可追溯到 1 年多以前的完整数据。

在 TV 中发现的大多数不匹配值的数据都少于 5 个月。

有人说需要先平滑数据，等等。到目前为止，我对这个脚本给出的数值还算满意。

参考：

https://stockcharts.com/school/doku.php?id=chartschool:technicalindicators:relativestrengthindex_rsi
https://stackoverflow.com/questions/22626238/calculate-rsirelative-strength-index-using-some-programming-language-js-c