Search code examples
pythonvectornumpy

Making a 0 to n vector in python


I am a new python user and I was wondering how I could make a 0 to n vector. I would like the user to be able to input an integer for n, and receive an output of [0,1,2,3,4,5...,n].

This is what I have done so far...

from numpy import matrix

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

But this is what I get as an output:

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

Transposing the matrix doesn't help. What am I doing wrong?

Thank you for your help!


Solution

  • Use the built-in function:

    range(n)
    

    (Well, should be n+1 if you want a list to be [0, 1, ... , n])