2017-04-01 60 views
0

这是一个非常简单的question.My的目标是最终建立一个Vandermonde矩阵,但首先我需要:简单Vandermonde矩阵与循环

编写一个函数,需要一个实数α和一个整数n作为输入并返回一个矢量v =(1,α,α2,...,αN-1)

到目前为止我有:

进口numpy的作为NP

n =6 
a= 3 
for i in range(n): 
    v = np.array([1, a**2, a**(n-1)]) 
print v 

我得到:[1 9 243]这不是我想要的。请你帮忙,谢谢!

回答

1

使用列表理解

v = np.array([a**x for x in range(n)]) 

这相当于:

v = [] 
for x in range(n): 
    v.append(a**x) 
v = np.array(v) 
+0

这是完美非常感谢你! – Kate

+0

@凯特,不客气! – ForceBru