np.array([1, 0, 0], [0, 1, 0], [0, 0, 1]) code example

Example: how to import numpy array in python

>>> import numpy as np
>>> a = np.array([0, 1, 2, 3])
>>> a
array([0, 1, 2, 3])