在python中创建一个0到n的向量

时间:2012-07-23 03:42:49

标签: python vector numpy

我是一个新的python用户,我想知道如何制作一个0到n的向量。 我希望用户能够为n输入一个整数,并接收[0,1,2,3,4,5 ...,n]的输出。

这是我到目前为止所做的......

from numpy import matrix

n=int(raw_input("n= "))
for i in range(n, 0, -1):
K = matrix(i)
print K

但这是我得到的结果:

[0][1][2][3][4][5]...[n]

转置矩阵并没有帮助。 我做错了什么?

感谢您的帮助!

3 个答案:

答案 0 :(得分:12)

使用内置功能:

range(n)

(如果你想要一个列表为[0,1,...,n],那么应该是n + 1)

答案 1 :(得分:6)

如果您想使用numpy,可以使用arange

>>> import numpy as np
>>> np.arange(10)
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

答案 2 :(得分:1)

from numpy import array
n = int(raw_input("n= "))
k = array(range(n+1))
print k