[Python源码] 通达信,同花顺中国式SMA指标,EMA指数平均,MACD,RSI的最简python代码实现

通达信,同花顺指标中的中国式SMA,EMA的python实现

同花顺，通达信指标计算有个非常重要的函数SMA,和EMA，下面仅仅用python库panadas和numpy的内置函数实现

1. import numpy as np; import pandas as pd
   
2. 
   
3. def EMA(S,N):     #为了精度 S>4*N  EMA至少需要120周期      
   
4.     return pd.Series(S).ewm(span=N, adjust=False).mean().values   
   
5. 
   
6. def SMA(S, N, M=1):    #中国式的SMA,至少需要120周期才精确        
   
7.     K = pd.Series(S).rolling(N).mean()    #先求出平均值
   
8.     for i in range(N+1, len(S)):  K[i] = (M * S[i] + (N -M) * K[i-1]) / N  # 因为要取K[i-1]，所以 range(N+1, len(S))        
   
9.     return K

复制代码

EMA实现的很成功，pandas内置函数，速度快，SMA需要循环才能实现，不是最优的解决方案,下面采用pandas的内置函数来实现！

1. def SMA(S, N, M=1):        # 1）高效的方法
   
2.     return pd.Series(S).ewm(alpha=M/N, adjust=True).mean().values  
   
3. 
   
4. def SMA(S, N, M=1):        # 2）高效写法2
   
5.     return pd.Series(S).ewm(com=N-M, adjust=True).mean().values  
   
6. 
   
7. def SMA(S, N, M=1):        # 3）高效写法3
   
8.     return pd.Series(S).ewm(span=2N/M-1, adjust=True).mean().values  
   
9. 
   
10. def SMA(S, N, M=1):        # 4）python2的pandas没有ewm,用ewma实现
    
11.     return pd.ewma(S,com=N-M,adjust=True)  
    
12. 
    
13. #以上4种写法得到的数值是一样的

复制代码

有了这两个关键函数，就可以用它们构造出MACD指标和RSI指标，代码非常简单

1. def MACD(CLOSE,SHORT=12,LONG=26,M=9):    # EMA的关系，S取120日，和雪球小数点2位相同
   
2.     DIF = EMA(CLOSE,SHORT)-EMA(CLOSE,LONG);  
   
3.     DEA = EMA(DIF,M);      MACD=(DIF-DEA)*2
   
4.     return DIF,DEA,MACD
   
5.    
   
6. def RSI(CLOSE, N=24):      
   
7.     DIF = CLOSE-REF(CLOSE,1)
   
8.     return SMA(MAX(DIF,0), N) / SMA(ABS(DIF), N) * 100   

复制代码

以上函数实现的macd,rsi指标和雪球，同花顺，通达信等软件对照完全一致

very good

感謝分享