Creating a multidimensional array
Now that we know how to create a vector, we are ready to create a multidimensional NumPy array. After we create the matrix, we will again want to display its shape (see the arrayattributes.py
file in the Chapter02
folder of this book's code bundle), as shown in the following code snippets:
To create a multidimensional array, see the following code:
In: m = array([arange(2), arange(2)]) In: m Out: array([[0, 1],[0, 1]])
To display the array shape, see the following lines of code:
In: m.shape Out: (2, 2)
We created a 2 x 2 array with the arange()
function. Without any warning, the array()
function appeared on the stage.
The array()
function creates an array from an object that you give to it. The object needs to be array-like, for instance, a Python list. In the preceding example, we passed in a list of arrays. The object is the only required argument of the array()
function. NumPy functions tend to have a lot of optional arguments with predefined defaults.