How do I create a custom 4x4 array using NumPy?

Question

I am new to Python and I am having a bit of trouble with the array functions. I want to make a 4 by 4 array which contains the numbers from 1 to 16.

I know that using np.zeros((4,4)) outputs a 4x4 array with all zeros. Using np.array(range(17)) I can get an array of the required numbers BUT not in the correct shape (4x4).

It must be fairly simple, surely? All comments are much appreciated.

Answer 1

The problem is that you are creating an array of 17 values (zero through 16), which can't be reshaped to 4x4. Instead:

>>> a = np.arange(1, 17).reshape(4,4)
>>> a
array([[ 1,  2,  3,  4],
       [ 5,  6,  7,  8],
       [ 9, 10, 11, 12],
       [13, 14, 15, 16]])