如何获取numpy.array的维数

Question

我对学习python非常陌生，我正尝试在np. array n x m 中缩放矩阵。 问题是，如果将np.array中的矩阵作为输入，而我不知道该矩阵的范围有多大，我如何初始化大小 m ？ Python中是否有某些features可以用于此目的？

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]