Как получить размеры numpy.array

Question

Я очень новичок в изучении Python и пытаюсь масштабировать матрицу в np. array nxm .вопрос в том, что если в качестве входных данных задана матрица в np.array, и я не знаю, насколько велик диапазон матрицы, как я могу инициализировать размер m ?Существуют ли определенные значения features в Python, которые можно использовать для этого?

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)

Это arrayB просто для примера

Ожидаемый результат:

3

Answer 1

arr.shape - это то, что вы ищете, оно дает вам размеры массива nD.

В вашем случае вы хотите arr.shape[1]