如何检索numpy.array的维度

Posted 2021-03-29

我是学习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

答案

arr.shape正是您所寻找的，它为您提供了nD阵列的尺寸。

在你的情况下，你想要arr.shape[1]