如何检索 numpy.array 的维度

Question

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

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]