簡單Vandermonde矩陣與循環

Question

這是一個非常簡單的question.My的目標是最終建立一個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]這不是我想要的。請你幫忙，謝謝！

Answer 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)