如何获取numpy.array的维数

Question

I am very new to learning python and I am trying to scale a matrix in np. array np. array nxm . the question is, if a matrix in the np.array is given as input and I don't know how big the range the matrix, how I can initialize the size m ? Are there certain features in Python that can be used for this?

import numpy as np
def scaleArray(arr: np.array);
    ##how ??

arrayB = np.array([[1,2,4],
                   [3,4,5],
                   [2,1,0],
                   [0,1,0]])
scaleArray(b)

this arrayB is just for example

Expected output :

3

Answer 1

arr.shape is what you are looking for, it gives you the dimensions of the nD array.

In your case, you want arr.shape[1]