简单的Vandermonde矩阵与for循环

Question

这是一个非常简单的问题。我的目标是最终创建一个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)