Numpy Arange

import numpy as np

a = np.arange(10)

print(a)

[0 1 2 3 4 5 6 7 8 9]

a.shape

(10,)

a.ndim

1

# Reshape

b = np.arange(12)

print(b)

[ 0  1  2  3  4  5  6  7  8  9 10 11]

b.shape

(12,)

print(b.ndim)

1

b.shape = (2, 3, 2)

print(b)

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

 [[ 6  7]
  [ 8  9]
  [10 11]]]

b.shape = (12, 1)

print(b)

[[ 0]
 [ 1]
 [ 2]
 [ 3]
 [ 4]
 [ 5]
 [ 6]
 [ 7]
 [ 8]
 [ 9]
 [10]
 [11]]

b.reshape(4, 3)

array([[ 0,  1,  2],
       [ 3,  4,  5],
       [ 6,  7,  8],
       [ 9, 10, 11]])

print(b)

[[ 0]
 [ 1]
 [ 2]
 [ 3]
 [ 4]
 [ 5]
 [ 6]
 [ 7]
 [ 8]
 [ 9]
 [10]
 [11]]

b = b.reshape(6, 2)

print(b)

[[ 0  1]
 [ 2  3]
 [ 4  5]
 [ 6  7]
 [ 8  9]
 [10 11]]