通达信一些指标的Python实现

对照通达信一些指标的Python实现

1. 
   
2. # 威廉指标
   
3. def williams(df, n, column='williams'):
   
4. # 100*(10日内最高价的最高值-收盘价)/(10日内最高价的最高值-10日内最低价的最低值)
   
5. for i in range(len(df)):
   
6. if i < n-1: continue
   
7. df.ix[i, column] = 100 * (df.high.values[i-n+1:i+1].max()-df.close.values[i])/(
   
8. df.high.values[i-n+1:i+1].max()-df.low.values[i-n+1:i+1].min())
   
9. return df
   
10. 
    
11. # 布林指标
    
12. def bollinger(df,n):
    
13. for i in range(len(df)):
    
14. if i < n-1: continue
    
15. df.ix[i, 'BOLL'] = df.close.values[i-n+1:i+1].mean()
    
16. df.ix[i, 'UB'] = df.ix[i, 'BOLL'] + 2 * numpy.std(df.close.values[i-n+1:i+1], ddof=1)
    
17. df.ix[i, 'LB'] = df.ix[i, 'BOLL'] - 2 * numpy.std(df.close.values[i-n+1:i+1], ddof=1)
    
18. return df
    
19. 
    
20. # 轨道线
    
21. def ene(df,n,m1,m2):
    
22. for i in range(len(df)):
    
23. if i < n-1: continue
    
24. df.ix[i, 'UPPER'] = (1+m1/100)*df.close.values[i-n+1:i+1].mean()
    
25. df.ix[i, 'LOWER'] = (1-m2/100)*df.close.values[i-n+1:i+1].mean()
    
26. df.ix[i, 'ENE'] = (df.ix[i, 'UPPER'] + df.ix[i, 'LOWER'])/2
    
27. return df
    
28. 
    
29. def kdj(df,n,m1,m2):
    
30. for i in range(len(df)):
    
31. if i < n-1: continue
    
32. df.ix[i, 'rsv'] = (df.close.values[i]-df.low.values[i-n+1:i+1].min()) / (df.high.values[i-n+1:i+1].max()-df.low.values[i-n+1:i+1].min())*100
    
33. df = getSMA(df,m1,1,'rsv','K')
    
34. df = getSMA(df,m2,1,'K','D')
    
35. for i in range(len(df)):
    
36. df.ix[i, 'J'] = 3*df.K.values[i] - 2*df.D.values[i]
    
37. return df
    

复制代码

行情软件里习惯了通达信的简介，虽然很多时候还是要依赖大智慧，但平时看图形基本都是用通达信。因此在有时候做数据分析的时候，不可避免的需要再次的实现一些指标功能，所以在Python里整理了一下，写了部分的指标工具。所有数据都是Dataframe类型，以时间为Index顺序排列。关于getSMA这个函数可以在本博另一帖子《

行情软件里的平均函数以及Python的实现》中可以找到。