关于二维数组的运算，贴段KDJ指标的代码，在8楼

hlsd665

有如下长度为N的数组
X＝［
［5.12，5.13，5.25，5.75］
［5.42，5.33，5.75，5.55］
［5.32，5.83，5.25，5.23］
［5.92，5.23，5.05，5.75］
［5.22，5.03，5.25，5.95］
    。
    。
    。
］

数组中的数值之间没有任何关系，
循环（len(x)-M）次
从第一个元素X［0］起取M个元素，做计算，
然后从第第二个元素X［1］起取M个元素，做计算。
以此类推，形式上用普通的循环就可以做到，


我没有学过高等数学，想问下有没有更简单的解法做优化。

Kabie

M会大于子元素的长度吗。。。

iamkey9

回复 1# hlsd665

numpy, array, tril, triu,

this may help.
http://docs.scipy.org/doc/numpy/ ... array-creation.html

hlsd665

回二楼：M小于数组的长度，
回三楼：谢谢您。我去看下。

感谢三楼，从您的指引下找到了一本好书“用Python做科学计算“，真不错啊，

地址如下：http://pyscin.appspot.com/html/index.html

iamkey9

回二楼：M小于数组的长度，
回三楼：谢谢您。我去看下。

感谢三楼，从您的指引下找到了一本好书“用Pyt ...
hlsd665 发表于 2011-01-13 07:21

这本书本版有人介绍过，你向作者发邮件，他会给你PDF版本，比网页新，不过作者要求不能共享

hlsd665

回5楼：
其实我用的部分只有数组处理部分，

作者说了，这本书适合工科、本科。。。。

我的数学只有高中水平，所以只取对我有用的部分，

另外再次感谢您。

Kabie

这不就是

1. for i in range(len(x)-M):
   
2.     XXX(x[i:i+M])

复制代码

你还打算怎么简化呢。。。。

hlsd665

横向可以切片，纵向不可以切片，在用引入NUMPY模块之前。

取出来的值用来画了KDJ指标。

1. import numpy as np
   
2. 
   
3. 
   
4. def KDJ(date,N=9,M1=3,M2=3):
   
5.     datelen=len(date)
   
6.     array=np.array(date)
   
7.     kdjarr=[]
   
8.     for i in range(datelen):
   
9.         if i-N<0:
   
10.             b=0
    
11.         else:
    
12.             b=i-N+1
    
13.         rsvarr=array[b:i+1,0:5]
    
14.         rsv=(float(rsvarr[-1,-1])-float(min(rsvarr[:,3])))/(float(max(rsvarr[:,2]))-float(min(rsvarr[:,3])))*100
    
15.         if i==0:
    
16.             k=rsv
    
17.             d=rsv
    
18.         else:
    
19.             k=1/float(M1)*rsv+(float(M1)-1)/M1*float(kdjarr[-1][2])
    
20.             d=1/float(M2)*k+(float(M2)-1)/M2*float(kdjarr[-1][3])
    
21.         j=3*k-2*d
    
22.         kdjarr.append(list((rsvarr[-1,0],rsv,k,d,j)))
    
23.     return kdjarr

复制代码